torch.cat()函数的理解

torch.cat() 是 PyTorch 中的一个函数，用于在指定的维度上连接（concatenate）张量（tensors）。它的功能是将多个张量沿着指定的维度拼接在一起。

函数签名如下：

torch.cat(tensors, dim=0, *, out=None) -> Tensor

• tensors: 一个包含要连接的张量的可迭代对象。
• dim: 指定连接的维度。例如，如果是 0，则在第一个轴上连接；如果是 1，则在第二个轴上连接，依此类推。
• out（可选）: 输出张量，用于指定结果的存储位置。如果未提供，将创建一个新的张量来存储结果。

简单的示例：

import torch

# 两个张量
tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6]])

# 在第一维度上连接（按行连接）
result1 = torch.cat((tensor1, tensor2), dim=0)
print(result1)
# Output:
# tensor([[1, 2],
#         [3, 4],
#         [5, 6]])

# 在第二维度上连接（按列连接）
result2 = torch.cat((tensor1, tensor2.T), dim=1)
print(result2)
# Output:
# tensor([[1, 2, 5],
#         [3, 4, 6]])