How can I stack tensors like this?

111226 · January 20, 2020, 2:02am

I’m trying to stack tensors with different size. But only dimension 2 is different while other dimensions are all same. Is there any way to do this?

ptrblck · January 20, 2020, 2:28am

torch.cat should work:

a = torch.randn(5, 5, 1)
b = torch.randn(5, 5, 4)
c = torch.cat((a, b), dim=2)
print(c.shape)
> torch.Size([5, 5, 5])

111226 · January 20, 2020, 4:00am

thanks bro, you saved me