pytorch如何将两个张量直接拼接成一个大张量

可以使用 torch.cat() 函数将两个张量直接在某个维度上进行拼接，生成一个大张量。例如，使用以下代码可以将两个大小为 (3, 4) 和 (3, 5) 的张量在第二个维度上进行拼接：

import torch

x = torch.randn(3, 4)
y = torch.randn(3, 5)

z = torch.cat([x, y], dim=1)  # 在第二个维度上进行拼接

print(z.size())  # 输出: torch.Size([3, 9])

在这里，`torch.cat([x, y], dim=1)` 的第一个参数是要拼接的所有张量，第二个参数 `dim=1` 表示要在第二个维度上进行拼接。拼接后生成的张量大小为 (3, 9)。