Tensorflow tf.tile()用法

tf.tile()应用于需要张量扩展的场景,具体说来就是: 
如果现有一个形状如[widthheight]的张量,需要得到一个基于原张量的,形状如[batch_size,width,height]的张量,其中每一个batch的内容都和原张量一模一样。tf.tile使用方法如:

tile(
    input,
    multiples,
    name=None
)
import tensorflow as tf
a  = tf.constant([7,19])
a1 = tf.tile(a,multiples=[3]) #第一个维度扩充3遍
b = tf.constant([[4,5],[3,5]])
b1 = tf.tile(b,multiples=[2,3])#第一个维度扩充2遍,第二个维度扩充3遍
with tf.Session() as sess:
    print(sess.run(a))
    print(sess.run(a1))
    print(sess.run(b))
    print(sess.run(b1))

猜你喜欢

转载自blog.csdn.net/qq_38826019/article/details/81588788