pytorch笔记:topk

1 介绍 

  • 返回给定输入张量沿指定维度的最大的 k 个元素

  • 如果没有指定 dim 参数,则默认选择输入张量的最后一个维度
  • 如果 largest 设置为 False,则返回最小的 k 个元素。
  • 返回一个命名元组 (values, indices),其中包含每一行在指定维度 dim 上最大的 k 个元素的值和索引。
  • 如果布尔选项 sorted 设置为 True,则确保返回的 k 个元素是有序的。

2 参数

input (Tensor)输入张量
k (int)表示“top-k”的 k 值
dim (int, 可选)要排序的维度
largest (bool, 可选)控制是返回最大还是最小的元素
sorted (bool, 可选)控制是否以排序的顺序返回元素。

3 举例

import torchx=torch.arange(12).reshape(2,6)x
'''
tensor([[ 0,  1,  2,  3,  4,  5],[ 6,  7,  8,  9, 10, 11]])
'''torch.topk(x,3)
'''
torch.return_types.topk(
values=tensor([[ 5,  4,  3],[11, 10,  9]]),
indices=tensor([[5, 4, 3],[5, 4, 3]]))
'''torch.topk(x,3).indices
'''
tensor([[5, 4, 3],[5, 4, 3]])
'''

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/diannao/20382.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Frida

配置环境 查看ubuntu版本 cat /etc/lsb-release设置时区 dpkg-reconfigure tzdataadb adb:(Android Debug Bridge)是一个用于在计算机和Android设备之间进行通信的命令行工具。它允许开发人员在计算机上执行各种命令,如安装应…

Git配置SSH-Key

git config --global user.name 沈健 git config --global user.email sjshenjianoutlook.com初次使用 SSH 协议进行代码克隆、推送等操作时,需按下述提示完成 SSH 配置 1 生成 RSA 密钥 ssh-keygen -t rsa2. 获取 RSA 公钥内容,并配置到 SSH公钥 中 …

FENDI CLUB精酿啤酒中原麦汁浓度的高低有何区别?

关于精酿啤酒,有两个关键数据,一个是原麦汁浓度,一个是酒精度。酒精度无非是含酒精的高低,但原麦汁浓度又是什么呢?另外精酿啤酒中原麦汁浓度有高有低,究竟有哪些区别呢? 原麦汁浓度是指啤…

c语言是编程软件还是编程语言?深入解析C语言的本质与定位

c语言是编程软件还是编程语言?深入解析C语言的本质与定位 在科技日新月异的今天,编程语言和编程软件在软件开发中扮演着举足轻重的角色。然而,对于初学者来说,这两者之间的界限往往模糊不清,尤其是对于像C语言这样的经…

息壤,让全网算力随愿可得

【全球云观察 | 科技热点关注】 当下的AIGC时代,算力领域面临的最大挑战是什么? 对于这个问题,仁者见仁,智者见智。但答案中最吸引业界眼球的莫过于,算力调度。为什么呢? 「因时而生」&#xf…

2006NOIP普及组真题 4. 数列

线上OJ: 【06NOIP普及组】数列 思考: 这道题大概率是一道可以使用“瞪眼法”找到规律的题目。我们尝试把数据补充的更多,以便于寻找规律 当 k3 时,k的幂次为1, 3, 9, 27, 81… 从上述推理中,我们发现要输出的幂次和中…

Flutter课程分享 -(系统课程 基础 -> 进阶 -> 实战 仿京东商城)

前言 在移动应用开发的世界中,Flutter 作为一款由 Google 推出的开源 UI 软件开发工具包,正迅速赢得开发者们的青睐。其跨平台、高性能、丰富的组件库以及易于学习的特性,使得 Flutter 成为许多开发者的不二选择。然而,对于初学者…

anaconda python 版本对应关系

如何创建python版本的虚拟环境 conda create -n d2l python3.5输入如下命令即可进入虚拟环境 activate d2lanaconda3版本基础python版本Anaconda3-2024.02-1Python 3.11.7Anaconda3-2023.09-0Python 3.11.5Anaconda3-2023.07-2Python 3.11.4Anaconda3-2023.07-1Python 3.11.3…

纯js仿淘宝多图片封面图插件模板/带视频,带放大镜,带前后端完整代码PHP

功能预览,他依赖jq插件,请自已引入 类似这样 <script type"text/javascript" src"/Application/Admin/Static/js/jquery-2.0.3.min.js"></script>一,前端模板代码 <!--多图功能--><style> charset "utf-8"; .wrap_imgs…

Javaweb基础之工程路径

大家好&#xff0c;这里是教授.F 引入&#xff1a; 工程路径有一个知识点需要注意&#xff1a;就是相对路径。所谓相对路径就是依赖当前位置&#xff1a; 相对路径的定位依赖于当前位置或参考位置。 使用相对路径来解决&#xff0c; 一个非常重要的规则&#xff1a;页面所有的…

H6922 2.8C-40V (最低启动电压2.5V)升压BOOST恒压芯片 5V12V24V升压IC

H6922升压BOOST恒压芯片是一款2.8C-40V &#xff08;最低启动电压2.5V&#xff09;升压BOOST恒压芯片 5V12V24V升压IC 首先&#xff0c;H6922的宽输入电压范围&#xff08;2.8-40V&#xff09;和低启动电压&#xff08;最低2.5V&#xff09;使其能够适应不同复杂的电源环境。无…

[书生·浦语大模型实战营]——训练自己的小助手认知+应用部署到 OpenXLab+复现多模态微调

1.训练自己的小助手认知 微调后的回答&#xff1a; 微调前的回答&#xff1a; 2.应用部署到 OpenXLab 上传的自我认知模型 应用部署在OpenXLab&#xff08;比上次部署方便不少&#xff0c;文档写的更清楚了&#xff0c;棒棒&#xff09;,链接如下应用链接 3.复现多模态…

普通人想自学Python?没弄清楚这些,我劝你还是算了吧......

自学Python&#xff0c;为何你值得一试&#xff1f; 在这个瞬息万变的时代&#xff0c;我们常常听到这样的声音&#xff1a;“自学Python&#xff1f;一般人我还是劝你算了吧。” 这样的言论&#xff0c;仿佛是一堵高墙&#xff0c;阻挡着那些怀揣梦想、渴望探索的普通人。然…

C++模板元编程

C模板元编程 为什么需要模板函数&#xff1f; 避免重复写代码 模板函数定义 使用template <class T> 或者template <typename T>其中T是可以变成任何类型调用时候T会替换成需要的类型 twice<int>会将T替换成int template <class T> T twice(T t) {re…

vue3+ts+element-plus 标签红色波浪形(属性不全)

新建文件shims-element-plus.d.ts,具体配置如下 import {ElAutocomplete,ElBadge,ElBreadcrumb,ElBreadcrumbItem,ElButton,ElCard,ElCheckbox,ElCheckboxGroup,ElCol,ElCollapse,ElCollapseItem,ElColorPicker,ElConfigProvider,ElContainer,ElDatePicker,ElDialog,ElDivide…

谷粒商城实战(031 业务-秒杀功能2)

Java项目《谷粒商城》架构师级Java项目实战&#xff0c;对标阿里P6-P7&#xff0c;全网最强 总时长 104:45:00 共408P 此文章包含第315p-第p318的内容 秒杀上架 定时上架功能 EnableAsync 异步 EnableScheduling 定时调度 Configuration 配置类 创建上架定时任务类和方法 …

408数据结构-图的存储与基本操作 自学知识点整理

前置知识&#xff1a;图的基本概念 图的存储必须完整、准确地反映顶点集和边集的信息。根据不同图的结构和算法&#xff0c;采用不同的存储方式将对程序的效率产生相当大的影响&#xff0c;因此选取的存储结构应适合于待求解的问题。 图的存储 邻接矩阵法 所谓邻接矩阵存储&a…

每天一个数据分析题(三百四十三)

同环比是帮助阅读者了解维度项表现水平有没有随着时间的推移而得以增长提升&#xff0c;以下对于同环比的说法正确的是&#xff1f; A. 同环比计算方法一般适用于对长周期业务行为结果的观测 B. 若产品具备季节属性特征&#xff0c;则应优先使用同比计算方法 C. 上期值一般指…

利用OBS+VLC+ASS插件实现多路rtsp慢直播流轮播播放

利用OBS+VLC+ASS插件实现多路rtsp慢直播流轮播播放 解决方案速览软件安装第一次使用OBS界面(不是第一次使用可以跳过该步骤)添加VLC视频源利用OBS多个**场景**特性实现布置多个VLC-rtsp视频流场景轮播解决方案速览 利用OBS进行本地直播画面的构建。添加VLC视频源:安装完VLC后…

c++框架介绍

C源文件后缀名.cc/.cpp&#xff0c;头文件后缀名.hh/.hpp C&#xff1a;编译使用g命令 3.C输入输出流比c中的printf更加简化 4.日志系统&#xff1a;拿到一个新的程序&#xff0c;怎么读取借助于日志系统&#xff1f;&#xff1f;&#xff1f;&#xff1f;&#xff1f;&…