pytorch如何进行张量拼接

PyTorch中可以使用torch.cat()函数进行张量拼接，该函数将多个张量按照指定的维度进行拼接。例如，可以使用以下代码将三个张量沿着第0维进行拼接：

import torch

tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6], [7, 8]])
tensor3 = torch.tensor([[9, 10], [11, 12]])

concat_tensor = torch.cat((tensor1, tensor2, tensor3), dim=0)
print(concat_tensor)

输出结果为：

tensor([[ 1,  2],
        [ 3,  4],
        [ 5,  6],
        [ 7,  8],
        [ 9, 10],
        [11, 12]])