【torch】pytorch实现矩阵转置

更高维的用torch.transpose. 其实二维的用transpose也可以了。但是我之前不是不知道嘛。下面的代码是盲敲的，没再验证。如果出问题了，就去pytorch官网查API吧~【此篇终结

t = torch.rand(3, 4, 5, 6)
t_t = torch.transpose(t, 3, 2)  # 想要哪两个维度置换后面的两个整数就是那两个维度

---

最下面是之前咔咔咔的搞，其实矩阵转置直接用tensor.T就可以了。。。

te = torch.rand(2, 3)
te_per = te.T

 

想要用矩阵的乘法，然后根据公式的话，我是需要对其中一个矩阵进行转置的，然后就发现说的太多，又没有给分析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])