top K元素的查找

top K问题是一个经典的问题。

问题描述:输入n个整数,输出其中最小的k个元素,例如,输入 1,2,3,4,5,6,7,8 那么最小的4个元素就是1,2,3,4.

除了这个,top K问题还指:常遇到的一类问题是,在海量数据中找出出现频率最高的前K个数,或者从海量数据中找出最大的前K个数,这类问题通常称为“top K”问题,如:在搜索引擎中,统计搜索最热门的10个查询词;在歌曲库中统计下载率最高的前10首歌等等。


参考:https://blog.csdn.net/ohmygirl/article/details/7843945

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
topk()是PyTorch中的一个函数,用于返回输入张量中的前k个最大值或最小值。它的参数包括input(输入张量)、k(前k个)、dim(默认为输入张量的最后一个维度)、sorted(是否排序)和largest(是否返回最大值)。\[2\] 例如,如果我们有一个张量m=torch.arange(0,10),我们可以使用m.topk(3)来获取前3个最大值。输出将是一个torch.return_types.topk对象,其中包含values和indices两个张量。values张量包含前3个最大值,indices张量包含这些值在原始张量中的索引。\[2\] 在给定的代码示例中,partition()和select()函数是用来实现快速选择算法的。快速选择算法是一种用于在未排序的列表中查找第k个最小或最大元素的算法。它通过递归地将列表分成较小和较大的两部分,并根据主元的值来确定下一步的操作。\[3\] #### 引用[.reference_title] - *1* [python topk](https://blog.csdn.net/itnerd/article/details/103605986)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [Python torch.topk() 函数用法](https://blog.csdn.net/sweet_tea_/article/details/128554578)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [python TopK算法](https://blog.csdn.net/xingtianyao/article/details/101053321)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值