tf.tile函数

函数原型

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

函数说明

tile函数主要用于张量的扩充，参数multiples表示扩充的倍数。比如，对于一个二维张量，multiples=[2, 3]表示张量在维度一（行）上面扩充两倍，在维度二（列）上面扩充三倍。

函数使用

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