Tensorflow：沿轴的堆叠（m，n，k，p）和（m，n，1，p）张量= 2

Question

I have to stack (m, n, k, p) and (m, n, 1, p) tensors along the axis=2.

tf.pack can only manage tensors of the same shape.

I've tried to use tf.unpack but it also doesn't work.

Answer 1

What I believe you are trying to do is called 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)])