тензорный поток:: опс:: 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
может быть объединен с последующей операцией, а затем оптимизирован.)
Аргументы:
- область: объект области.
- input: Тензор для трансляции.
- shape: 1-D
int
Tensor . Форма желаемого результата.
Возврат:
Конструкторы и деструкторы | |
---|---|
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::Выход
operator::tensorflow::Output() const