pytorch的 transpose()的功能

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]])

猜你喜欢

转载自blog.csdn.net/Vertira/article/details/130644772