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

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

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

错误理解

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

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

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

Torch.argsort()“返回结果错误”的理解_第1张图片
但是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的意思

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

Torch.argsort()“返回结果错误”的理解_第2张图片

Torch.argsort()“返回结果错误”的理解_第3张图片

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

要点回顾

argsort 返回的不是rank,而是indcies

参考链接

pytorch issue

你可能感兴趣的:(pytorch)