【torch】pytorch实现矩阵转置

 想要用矩阵的乘法，然后根据公式的话，我是需要对其中一个矩阵进行转置的，然后就发现说的太多，又没有给分析permute的源码，所以就自己摸官网试了一下，如下代码是可以实现转置操作的mat2和mat3都是转置后的结果，所以我就不知道那个参数起了什么作用了，我也没心思去看源码，因为试验做不完了呀。

if __name__ == '__main__':
    mat1 = torch.randn(4, 5)
    print(mat1)
    mat2 = mat1.permute(1, 0)  # 转置
    print(mat2)
    mat3 = mat1.permute(-1, -2)
    print(mat3)

下面是刚刚看到的一种方式，源自https://zhuanlan.zhihu.com/p/44954540

a = torch.arange(6).reshape(2,3)
torch.einsum('ij->ji',[a])