繁体   English   中英

Tensorflow:沿轴的堆叠(m,n,k,p)和(m,n,1,p)张量= 2

[英]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.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM