Tensorflow2—tf.tile()

tf.tile()

函数原型

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

参数说明

• input ： 张量。 一维或更高
• multiples： 张量。 必须是以下类型之一：int32，int64。 一维长度必须与输入中的维度数相同
• name ： A name for the operation (optional).
  函数说明
  根据multiples张量复制 input 产生新的张量。 multiples 中指定 input 中每一个维度的复制次数

a = tf.constant([[1,2,3],[4,5,6]],tf.int32)
b = tf.tile(a,[1,2])
print(a)
print(b)
结果：
tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
tf.Tensor(
[[1 2 3 1 2 3]
 [4 5 6 4 5 6]], shape=(2, 6), dtype=int32)

显然，在行维度上复制了一次，列维度上复制了两次。

a = tf.constant([[1,2,3],[4,5,6]],tf.int32)
b = tf.tile(a,[2,1])
print(a)
print(b)
结果：
tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
tf.Tensor(
[[1 2 3]
 [4 5 6]
 [1 2 3]
 [4 5 6]], shape=(4, 3), dtype=int32)

在行维度上复制了两次，列维度上复制了一次。