关于pytorch的一些小技巧

对于一个二维矩阵,使用两次sort函数(对原矩阵、第一次sort之后的index)后,index(第二个变量)的输出是原矩阵一个由大到小(或由小到大)的映射,映射的值为整数,位置保持不变。 复原原始二维矩阵:a1(b11)=a 有意思,记录一下。
相关文章
相关标签/搜索