tf.broadcast_to

Распространить массив для совместимой формы.

tf.broadcast_to(
    input: Annotated[Any, TV_BroadcastTo_T],
    shape: Annotated[Any, TV_BroadcastTo_Tidx],
    name=None
) -> Annotated[Any, TV_BroadcastTo_T]

Используется в блокнотах

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

Например:

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 может быть объединен с последующей операцией и затем оптимизирован, однако.)