pytorch.cat()

import torch

# x1
x1 = torch.tensor([[11, 21, 31], [21, 31, 41]], dtype=torch.int)
print(x1.shape)  # torch.Size([2, 3])
# x2
x2 = torch.tensor([[12, 22, 32], [22, 32, 42]], dtype=torch.int)
print(x2.shape)  # torch.Size([2, 3])

inputs = [x1, x2]
print(inputs)

print(torch.cat(inputs, dim=2))

https://blog.csdn.net/xinjieyuan/article/details/105208352