I have two tensors. For example -
a = torch.randn((500, 200, 10)) b = torch.randn((500, 5))
I have to concat each of
b tensor to all elements of corresponding
a tensor i.e., each 200 tensors of
a should get concatenated with
b - final dimension should be
(500, 200, 15).
Without using explicit
for loop, how can I achieve this in Pytorch efficiently?