torch numpy sort排序出现索引顺序不正常

问题

torch 调用numpy ,numpy 的sort排序值的顺序没发现问题，但是排序的索引argsort()是有明显问题。

方案

多一次取索引

arr.argsort().argsort()

参考：

https://blog.youkuaiyun.com/qq_50571974/article/details/123173118

说明：argsort其实是没问题的，理解有点复杂，以序号的角度去测试和思考