在PyTorch中，可以使用torch.view()方法来改变Tensor的维度。torch.view()方法接受一个元组作为参数，该元组指定了新的维度。例如，如果要将一个形状为(2, 3)的Tensor改变为形状为(3, 2)的Tensor，可以使用以下代码：
云服务器
物理服务器
虚拟主机
跨境电商服务器import torch
# 创建一个形状为(2, 3)的Tensor
x = torch.tensor([[1, 2, 3],
[4, 5, 6]])
# 将Tensor改变为形状为(3, 2)
x_reshaped = x.view(3, 2)
print(x_reshaped)
除了torch.view()方法外，还可以使用torch.reshape()方法来改变Tensor的维度。torch.reshape()方法与torch.view()方法类似，但有一些细微的差别。例如，torch.reshape()方法会在需要的情况下对数据进行复制，而torch.view()方法不会。
要使用torch.reshape()方法改变Tensor的维度，可以使用以下代码：
import torch
# 创建一个形状为(2, 3)的Tensor
x = torch.tensor([[1, 2, 3],
[4, 5, 6]])
# 将Tensor改变为形状为(3, 2)
x_reshaped = torch.reshape(x, (3, 2))
print(x_reshaped)
无论是使用torch.view()方法还是torch.reshape()方法，都可以方便地改变Tensor的维度。
辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>
推荐阅读： PyTorch中怎么压缩模型