tensorflow::ops::BroadcastTo

#include <array_ops.h>

Передайте массив совместимой формы.

Краткое содержание

Широковещательная передача — это процесс создания массивов с совместимыми размерами для выполнения арифметических операций. Два массива считаются совместимыми, если для каждой пары измерений они либо равны, либо одно из них равно единице. При попытке широковещательной передачи тензора в массив, процесс начинается с последних измерений и продолжается дальше.

Например,

x = tf.constant([1, 2, 3]) y = tf.broadcast_to(x, [3, 3]) print(y) tf.Tensor( [[1 2 3] [1 2 3] [1 2 3]], shape=(3, 3), dtype=int32)

В приведенном выше примере входной тензор размером [1, 3] преобразуется в выходной тензор размером [3, 3] .

При выполнении операций с широковещательной передачей, таких как умножение тензора на скаляр, широковещательная передача (обычно) дает некоторые преимущества по времени или пространству, поскольку передаваемый тензор никогда не материализуется.

Однако broadcast_to не предоставляет подобных преимуществ. Вновь созданный тензор занимает всю память, содержащую передаваемую форму. (В контексте графа, broadcast_to может быть объединена с последующей операцией, а затем оптимизирована.)

Аргументы:

  • scope: Объект Scope
  • Входные данные: Тензор для трансляции.
  • shape: Одномерный int тензор . Форма желаемого выходного значения.

Возвращает:

Конструкторы и Деструкторы

BroadcastTo (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input shape)

Общедоступные атрибуты

operation
output

Общественные функции

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Общедоступные атрибуты

операция

Operation operation

выход

::tensorflow::Output output

Общественные функции

BroadcastTo

 BroadcastTo(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input shape
)

узел

::tensorflow::Node * node() const 

оператор::tensorflow::Input

 operator::tensorflow::Input() const 

оператор::tensorflow::Output

 operator::tensorflow::Output() const