tf.raw_ops.BroadcastTo

Транслирует массив к совместимой форме.

tf.raw_ops.BroadcastTo(
    input, shape, name=None
)

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

Например:

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

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

При трансляции, если у тензора меньше осей, чем требуется, его форма дополняется единицами слева. Поэтому следующий пример даёт тот же результат, что и предыдущий:

x = tf.constant([1, 2, 3])   # Shape (3,)
y = tf.broadcast_to(x, [2, 3])

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

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