How to concatenate tensor to another list of tensor in pytorch?

I have a tensor of shape "torch.Size([2, 2, 3])" and another tensor of shape "torch.Size([2, 1, 3])". I want a concatenated tensor of shape "torch.Size([2, 2, 6])". For example :

a=torch.tensor([[[2,3,5],[12,13,15]],[[20,30,50],[120,130,150]]]) b=torch.tensor([[[99,99,99]],[[999,999,999]]])

I want the output as : [[[99,99,99,2,3,5],[99,99,99,12,13,15]],[[999,999,999,20,30,50],[999,999,999,120,130,150]]]

I have written a O(n2) solution using two for loops but, This is taking a lot of time with millions of calculation, Does anyone help me in doing this efficiently ?? May be some matrix calculation trick for tensors ??

https://stackoverflow.com/questions/66739326/how-to-concatenate-tensor-to-another-list-of-tensor-in-pytorch March 22, 2021 at 09:37AM