Torch.argsort()“返回结果错误”的理解

Torch.argsort()“返回结果错误”

最近对于torch.argsort()理解不太到位
记录一下疑问以及正确解释

错误理解

之前误以为torch.argsort()返回的是

该元素的值的排名，比如[1,2,3]的argsort排名为[0,1,2]表示第一个元素在向量中排名为0

带着这个错误理解，去查阅torch.argsort官方文档：

但是example中第一行 -0.8521这个元素明显应该排在第一位也就是rank 0，但是给的值却是3.

那必定是哪里出了问题

问题解决

原来troch.argsort返回的并不是每个值的rank，而是

Returns the indices that sort a tensor along a given dimension in ascending order by value.

返回的是一个排序好的列表值的索引。

之前误解了argsort的意思

正确的理解以及使用方法是：

也就是根据所给的索引，依次取出元素，就会得到一个排序好的tensor。

要点回顾

argsort 返回的不是rank，而是indcies

参考链接

pytorch issue