用于在指定的维度上拼接tensor(张量)。
import torch
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]])