Как склеить тензоры в PyTorch?

Question

[iramovich]

Как мне в Pytorch склеить три тензора

tensor([[ 96, 20, 96, 21, 96, 960, 577, 2, 96, 21, 96, 21, 96]])
tensor([[37954, 2, 96, 20, 96, 21, 96]])
tensor([[16385, 2, 96, 21, 96, 21, 96]])

так чтобы получился следующий тензор:

tensor([[ 96, 20, 96, 21, 96, 960, 577, 2, 96, 21,
96, 21, 96, 37954, 2, 96, 20, 96, 21, 96,
16385, 2, 96, 21, 96, 21, 96]])

Answer 1

[Сергей Соколов]

import  torch 

a = torch.tensor([[ 96, 20, 96, 21, 96, 960, 577, 2, 96, 21, 96, 21, 96]])
b = torch.tensor([[37954, 2, 96, 20, 96, 21, 96]])
c = torch.tensor([[16385, 2, 96, 21, 96, 21, 96]])

torch.cat([a,b,c], 1)

###
tensor([[   96,    20,    96,    21,    96,   960,   577,     2,    96,    21,
            96,    21,    96, 37954,     2,    96,    20,    96,    21,    96,
         16385,     2,    96,    21,    96,    21,    96]])