1 128 20 20 1 64 40 40 1 128 40 40如何使用torch.cat函数对上面这三个维度的张量进行拼接（dimension=1）

在PyTorch中，`torch.cat()` 函数用于将一维或多维张量沿着指定维度进行连接。对于给定的三个二维张量：

1. `torch.tensor([1, 128, 20, 20])`
2. `torch.tensor([1, 64, 40, 40])`
3. `torch.tensor([1, 128, 40, 40])`

要按照第1维度（即列方向，因为索引从0开始计数，所以 dimension=1 对应于传统的列拼接）拼接它们，你可以这样做：

import torch

# 定义每个张量
tensor1 = torch.tensor([[1], [128], [20], [20]])
tensor2 = torch.tensor([[1], [64], [40], [40]])
tensor3 = torch.tensor([[1], [128], [40], [40]])

# 使用 torch.cat() 沿着第1维度进行拼接
result = torch.cat((tensor1, tensor2, tensor3), dim=1)

print(result)

运行这段代码后，你会得到一个5x4的新张量，其中包含了原始三个张量的所有行。