I have two tensors of shape [12, 39,1024] and [12, 39,1024]. I want to concatenate them depth-wise but in a one-on-one fashion. For example, the first feature map of the first tensor is attached to the first feature map of the second tensor. So the sequence would be [(Feature map of 1st) Concatenated with (Feature map of 2st) ]. If i use existing PyTorch concatenation operations like a torch.cat(), then it will simply concatenate the second tensor at the end. Is there any pytorch function which can help me achieve what i need?