作用

用于在指定的维度上拼接tensor（张量）。

导入

python">import torch

用法

python">tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6], [7, 8]])
result = torch.cat((tensor1, tensor2), dim=0)# result:
# tensor([[1, 2],
#        [3, 4],
#        [5, 6],
#        [7, 8]])