關於pytorch的一些小技巧

對於一個二維矩陣,使用兩次sort函數(對原矩陣、第一次sort之後的index)後,index(第二個變量)的輸出是原矩陣一個由大到小(或由小到大)的映射,映射的值爲整數,位置保持不變。 復原原始二維矩陣:a1(b11)=a 有意思,記錄一下。
相關文章
相關標籤/搜索