更高维的用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])