pytorch调换维度

以下是两种PyTorch调换维度的方法：

1.使用permute()函数进行维度交换

import torch

a = torch.rand(4, 3, 28, 32)
b = a.permute(0, 2, 3, 1) # 将第0维和第2维、第3维和第1维交换
print(b.shape) # 输出：torch.Size([4, 28, 32, 3])

2.使用transpose()函数进行维度交换

import torch

a = torch.tensor([1.2, 2.3])
b = a.transpose(0, 0) # 将第0维和第0维交换
print(b.shape) # 输出：torch.Size([2])