Python tf.broadcast_to用法及代码示例

广播一个兼容形状的数组。

用法

tf.broadcast_to(
    input, shape, name=None
)

广播是使数组具有用于算术运算的兼容形状的过程。如果每个维度对的两个形状相等或其中之一是一个，则两个形状是兼容的。当尝试将张量广播到一个形状时，它从尾随维度开始，然后向前推进。

例如，

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 没有任何此类好处。新创建的张量完全 memory 广播的形状。 (但是，在图形上下文中，broadcast_to 可能会融合到后续操作中，然后被优化掉。)