关于pytorch的一些小技巧


对于一个二维矩阵,使用两次sort函数(对原矩阵、第一次sort之后的index)后,index(第二个变量)的输出是原矩阵一个由大到小(或由小到大)的映射,映射的值为整数,位置保持不变。




复原原始二维矩阵:a1(b11)=a 有意思,记录一下。