pytorch中的transpose方法的作用是交换矩阵的两个维度,transpose(dim0, dim1) → Tensor,其和torch.transpose()函数作用一样。
torch.transpose():
>>> n
tensor([[2, 3],
[4, 5]])
>>> n1=n.transpose(0,1) # 效果不变
>>> n1
tensor([[2, 4],
[3, 5]])
>>> n2=n.transpose(-1,-2) # 效果不变
>>> n2
tensor([[2, 4],
[3, 5]])