關於pytorch的一些小技巧

對於一個二維矩陣，使用兩次sort函數（對原矩陣、第一次sort之後的index）後，index（第二個變量）的輸出是原矩陣一個由大到小（或由小到大）的映射，映射的值爲整數，位置保持不變。 復原原始二維矩陣：a1(b11)=a 有意思，記錄一下。

>>阅读原文<<