torch.cat()

要将一个大小为(2, 2)的PyTorch张量和一个大小为(1, 2)的张量拼接在一起，以形成一个新的大小为(3, 2)的张量，你可以使用PyTorch库中的torch.cat()函数。以下是一个示例代码：

import torch

# 假设你有一个大小为(2, 2)的张量 tensor1 和一个大小为(1, 2)的张量 tensor2
tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6]])

# 使用 torch.cat() 进行拼接，指定维度为 0，表示在行方向进行拼接
new_tensor = torch.cat((tensor1, tensor2), dim=0)

print(new_tensor)

在这个示例中，我们首先创建了两个大小不同的张量 tensor1 和 tensor2。然后，我们使用 torch.cat() 函数将它们在维度0上拼接起来，得到了新的大小为(3, 2)的张量 new_tensor。这里的 dim=0 表示在行方向上进行拼接。请注意，拼接的张量维度要保持一致，除了拼接维度外的其他维度应该是一样的。