PyTorch torch.cat() 函数：在第 1 维度上连接张量

torch.cat([x1, x2], dim=1) 是 PyTorch 中用于连接张量的函数，其中 dim=1 表示在第 1 维度上进行连接。这意味着将 x1 和 x2 在列方向上进行拼接。例如，如果 x1 和 x2 分别是 (2, 3) 和 (2, 4) 的张量，则 torch.cat([x1, x2], dim=1) 将返回一个 (2, 7) 的张量，将 x2 的列追加到 x1 的列的后面。