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 可能會融合到後續操作中，然後被優化掉。)