当前位置: 代码迷 >> 综合 >> Tensorflow tf.tile()用法
  详细解决方案

Tensorflow tf.tile()用法

热度:50   发布时间:2023-12-27 18:15:31.0

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))