PyTorch torch.cat() 函数：在指定维度连接张量 - 详细指南和示例

"torch.cat([x1,x2], dim=1)" 在 PyTorch 中使用 torch.cat() 函数，可以在指定维度上连接多个张量。"dim=1" 参数指定在第 1 个维度上进行连接，即在列方向上连接两个张量。\n\n\n举例说明：\n假设 x1 是一个大小为 [3, 2] 的张量，x2 是一个大小为 [3, 2] 的张量，即有 3 行 2 列。\n\nx1:\n[[1, 2],\n [3, 4],\n [5, 6]]\n\nx2:\n[[7, 8],\n [9, 10],\n [11, 12]]\n\n使用 torch.cat([x1, x2], dim=1) 将 x1 和 x2 在列的方向上连接，得到的结果是一个大小为 [3, 4] 的张量。\n\n结果：\n[[1, 2, 7, 8],\n [3, 4, 9, 10],\n [5, 6, 11, 12]]\n\n代码示例：\npython\nimport torch\n\nx1 = torch.tensor([[1, 2], [3, 4], [5, 6]])\nx2 = torch.tensor([[7, 8], [9, 10], [11, 12]])\n\nresult = torch.cat([x1, x2], dim=1)\nprint(result)\n\n\n输出：\n\ntensor([[ 1, 2, 7, 8],\n [ 3, 4, 9, 10],\n [ 5, 6, 11, 12]])\n