@iramovich

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

Как мне в 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]])
  • Вопрос задан
  • 39 просмотров
Решения вопроса 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]])
Ответ написан
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы