- permute可以对任意高维矩阵进行转置。
- transpose只能操作2D矩阵的转置(连续使用transpose可实现permute效果)。
import torch
image = torch.randn(1,16 ,256, 256)
image = image.permute(1,0,2,3)
print(image.shape)#torch.Size([16, 1, 256, 256])
import torch
image = torch.randn(1,16 ,256, 256)
image = image.permute(1,0,2,3)
print(image.shape)#torch.Size([16, 1, 256, 256])