pytorch数据如何拼接

在 PyTorch 中，可以使用 torch.cat() 函数将多个张量沿指定维度进行拼接。具体的用法如下：

import torch

# 创建两个张量
x1 = torch.randn(3, 4)
x2 = torch.randn(3, 4)

# 沿行维度拼接两个张量
x = torch.cat([x1, x2], dim=0)
print(x.shape)

# 沿列维度拼接两个张量
y = torch.cat([x1, x2], dim=1)
print(y.shape)

其中，dim 参数表示要沿哪个维度进行拼接。在上面的例子中，dim=0 表示沿行维度进行拼接，dim=1 表示沿列维度进行拼接。注意，要求拼接的张量在除了拼接维度之外的所有维度大小都相同。