1 连续两个argsort
返回张量中每个元素对应的排名
torch.argsort(torch.argsort(pred, dim=1, descending=True),
dim=1,
descending=False)
例子
使用一个argsort后得到的是张量中按列降序排序后的索引,
再使用一个argsort后是张量中每一个元素的排名。
例如第2行中,[0,4,1]升序后的索引为[1,2,0],每个元素的排名为[2,0,1]
![]()

文章讨论了在PyTorch中如何使用argsort函数对张量进行两次排序。第一次argsort按列降序排列,返回索引;第二次argsort则给出每个元素在新序列中的排名。举例说明,如第二行[0,4,1]经过升序排序后变为[1,2,0],对应的排名为[2,0,1]。
2148

被折叠的 条评论
为什么被折叠?



