pytorch ：使用两次sort函数（排序）找出矩阵每个元素在升序或降序排列中的位置
2021-01-05 01:30
标签：操作 排名 sort函数 rank 表示 rand 索引 div 代码 在SSD的代码中经常有见到如下的操作： 其作用是什么呢？举个例子： 返回： 其实就是可以通过最后的rank看出对应位置的值的排序位置，这里是降序，所以索引0表示最高 以rank的第一行第一列的值0为例，其表示对应的a中第一行第一列的值2.3326是第一行中最大的，因为设置了dim=1;第一行第三列的值3为例，其表示对应的a中第一行第三列的值-0.0799是第一行中最小的 起到了对应位置排名的作用 pytorch ：使用两次sort函数（排序）找出矩阵每个元素在升序或降序排列中的位置 标签：操作 排名 sort函数 rank 表示 rand 索引 div 代码 原文地址：https://www.cnblogs.com/wanghui-garcia/p/12982732.html _, idx = flt[:, :, 0].sort(1, descending=True)#大小为[batch size, num_classes*top_k]
_, rank = idx.sort(1)#再对索引升序排列，得到其索引作为排名rank
import torch
a = torch.randn(3,4)
print(a)
print()
i, idx = a.sort(dim=1, descending=True)
print(i)
print(idx)
print()
j, rank = idx.sort(dim=1)
print(rank)
tensor([[ 2.3326, 0.0275, -0.0799, 0.4156],
[-2.2066, 1.7997, -2.2767, 0.4704],
[-0.6980, 0.2285, 1.0018, -0.8874]])
tensor([[ 2.3326, 0.4156, 0.0275, -0.0799],
[ 1.7997, 0.4704, -2.2066, -2.2767],
[ 1.0018, 0.2285, -0.6980, -0.8874]])
tensor([[0, 3, 1, 2],
[1, 3, 0, 2],
[2, 1, 0, 3]])
tensor([[0, 2, 3, 1],
[2, 0, 3, 1],
[2, 1, 0, 3]])
文章标题：pytorch ：使用两次sort函数（排序）找出矩阵每个元素在升序或降序排列中的位置
文章链接：http://soscw.com/index.php/essay/40159.html