我有两个Tensor的形状[2，1，9]和[2，1，3]。我想连接跨越第三维交替（一次每4）。例如：
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]]]
我怎么能在pytorch做这个？
tkclm6bt1#
这样就可以了：
torch.concat([a.reshape((2, 1, 3, 3)), b.reshape(2, 1, 3, 1)], axis=-1).reshape((2, 1, -1))
可能有更聪明的方法来做这件事，但嘿，它的工作。
1条答案
按热度按时间tkclm6bt1#
这样就可以了：
可能有更聪明的方法来做这件事，但嘿，它的工作。