pytorch选出数据中的前k个最大（最小）值及其索引

选择最大值及其索引，大家都知道使用max(),argmax()函数。

那么如何返回前k个最大值呢，这在我们计算topK准确率的时候很有必要：

在torch中，我们可以使用sort函数来实现：

a, idx1 = torch.sort(data, descending=True)#descending为alse，升序，为True，降序
idx = idx1[:k]

Return:
a:排好序的数据
idx1：对应排序数据的索引

因此只需设置k的大小，就可以截取到前k个最大值的索引。这里若数据是tensor则用torch，若是list或ndarray，可以用numpy。

pytorch选出数据中的前k个最大（最小）值及其索引相关推荐

网易_在数组中查找前K个元素
笔试题,最后一题查找网易云音乐中播放量最大的前K个歌曲. 换句话说,就是在数组中查找前K大元素. 大致有以下几个思路. 1.第一感觉就是对数组进行降序全排序,然后返回前K个元素,即是需要的K个最大数 ...
C语言编程＞第二十六周 ① 函数fun的功能是：将形参b所指数组中的前半部分元素的值和后半部分元素的值对换。形参n中存放数组中数据的个数，若n为奇数，则中间的元素不动。
例题:函数fun的功能是:将形参b所指数组中的前半部分元素的值和后半部分元素的值对换.形参n中存放数组中数据的个数,若n为奇数,则中间的元素不动. 例如,若a所指数组中的数据依次为:11 22 33 ...
leetcode 703. 数据流中的第K大元素最小堆解法 c语言
如题: 设计一个找到数据流中第K大元素的类(class).注意是排序后的第K大元素,不是第K个不同的元素. 你的 KthLargest 类需要一个同时接收整数 k 和整数数组nums 的构造器,它包含 ...
mysql数据中包含不间断空格（ascii值为194和160）解决办法
mysql数据中包含不间断空格(ascii值为194和160)解决办法问题 mysql数据存在这个不间断空格,导致前端显示乱码. 起因字符串中的一个空格(ASCII:32)被UTF-8编码之后变成 ...
Excel表格中如何找出并替换数据中某一固定范围内的值（包括空值）
Excel表格中如何找出并替换数据中某一固定范围内的值(包括空值) 一.问题介绍二.方法介绍 1.选中数据 2.按住Ctrl+H键,调出替换窗口 3.输入 " * ",并点击查找 ...
通达信公式：如何表示5个数据中的前三大数值？
因为5个数据每天都是变化的,因此需要相互比较,通达信里面的函数只能求最大和最小值,请问有没有大佬指点下如何输出第二大和第三大的数值?谢谢~
sql查找数据中某个字段是否有重复的值
表格table_x table_x col1 Column2 二仙桥谭警官二仙桥大爷瓜摊华强瓜摊瓜水煮肉片肉水煮肉片水水煮肉片豆芽菜水煮肉片土豆片水煮肉片花椒水煮肉 ...
77. Leetcode 1439. 有序矩阵中的第 k 个最小数组和 (堆-技巧二-多路归并)
技巧二 - 多路归并其实这个技巧,叫做多指针优化可能会更合适,只不过这个名字实在太过朴素且容易和双指针什么的混淆,因此我给 ta 起了个别致的名字 - 多路归并.多路体现在:有多条候选路线.代码上, ...
（补）算法训练Day13 | LeetCode150. 逆波兰表达式求值（栈应用）；LeetCode239. 滑动窗口最大值（单调队列）；LeetCode347. 前K个高频元素（小顶堆，优先级队列）
目录 LeetCode150. 逆波兰表达式求值 1. 思路 2. 代码实现 3. 复杂度分析 4. 思考 LeetCode239. 滑动窗口最大值 1. 思路 2. 代码实现 3. 复杂度分析 4. ...

pytorch选出数据中的前k个最大（最小）值及其索引

pytorch选出数据中的前k个最大（最小）值及其索引相关推荐

最新文章

热门文章