Python tf.tile用法及代碼示例

通過平鋪給定的張量來構造張量。

用法

tf.tile(
    input, multiples, name=None
)

此操作通過複製 input multiples 次來創建一個新的張量。輸出張量's i'th 維度具有input.dims(i) * multiples[i] 元素，並且input 的值沿'i'th 維度複製multiples[i] 次。例如，通過 [2] 平鋪 [a b c d] 會產生 [a b c d a b c d] 。

a = tf.constant([[1,2,3],[4,5,6]], tf.int32)
b = tf.constant([1,2], tf.int32)
tf.tile(a, b)
<tf.Tensor:shape=(2, 6), dtype=int32, numpy=
array([[1, 2, 3, 1, 2, 3],
       [4, 5, 6, 4, 5, 6]], dtype=int32)>
c = tf.constant([2,1], tf.int32)
tf.tile(a, c)
<tf.Tensor:shape=(4, 3), dtype=int32, numpy=
array([[1, 2, 3],
       [4, 5, 6],
       [1, 2, 3],
       [4, 5, 6]], dtype=int32)>
d = tf.constant([2,2], tf.int32)
tf.tile(a, d)
<tf.Tensor:shape=(4, 6), dtype=int32, numpy=
array([[1, 2, 3, 1, 2, 3],
       [4, 5, 6, 4, 5, 6],
       [1, 2, 3, 1, 2, 3],
       [4, 5, 6, 4, 5, 6]], dtype=int32)>