作用
用于在指定的维度上拼接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]])