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()操作的区别。