Alternate concatenation of tensors

1

I have 2 tensors of shape [2, 1, 9] and [2, 1, 3]. I’d like to concatenate across the 3rd dimension alternatively (once every 4).

For example:

a = [[[1,2,3,4,5,6,7,8,9]],[[11,12,13,14,15,16,17,18,19]]]
b = [[[10, 20, 30]], [[1, 2, 3]]]
result = [[[1,2,3,10,4,5,6,20,7,8,9,30]],[[11,12,13,1,14,15,16,2,17,18,19,3]]]

How can I do this in pytorch?

2

Solved by this

torch.concat([a.reshape((2, 1, 3, 3)), b.reshape(2, 1, 3, 1)], axis=-1).reshape((2, 1, -1))