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]