torch.transpose()函数

torch.transpose(Tensor,dim0,dim1)是pytorch中的ndarray矩阵进行转置的操作

注意：transpose()一次只能在两个维度间进行转置（也可以理解为维度转换）

例如：

import torch as t

x=t.Tensor(2,3,4,5) #这是一个4维的矩阵（只用空间位置，没有数据）
print(x.size())
#先转置0维和1维，之后在第2，3维间转置，之后在第1，3间转置
y=x.transpose(0,1).transpose(3,2).transpose(1,3)
print(y.size())

#输出：

torch.Size([2, 3, 4, 5])
torch.Size([3, 4, 5, 2])

Process finished with exit code 0

注意和torch.t()操作的区别。