[英]Tensorflow: stack (m, n, k, p) and (m, n, 1, p) tensors along the axis=2
我必须沿轴= 2堆叠(m,n,k,p)和(m,n,1,p)张量。
tf.pack
只能管理相同形状的张量。
我尝试使用tf.unpack
但它也不起作用。
我相信您要尝试做的就是称为tf.concat :
a = tf.zeros([3,4,10,5])
b = tf.zeros([3,4,1,5])
c = tf.concat(2, [a,b])
c.get_shape()
Out: TensorShape([Dimension(3), Dimension(4), Dimension(11), Dimension(5)])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.