当前位置：首页 > news >正文

PyTorch topk() 用法详解：取最大值

news 来源：原创 2025/7/3 18:33:03

torch.topk(input, k) 返回张量中最大的 k 个元素以及它们在原张量中的索引。

函数原型

torch.topk(input, k, dim=None, largest=True, sorted=True)

参数说明：

参数	说明
`input`	输入张量
`k`	要取出的前 k 个值
`dim`	指定沿哪个维度取值（默认是最后一维）
`largest`	是否取最大值（默认是 `True`，为 `False` 时返回最小值）
`sorted`	返回的结果是否排序（默认是 `True`，按值从大到小）

示例：二维张量中使用 `topk()`（`dim=0` vs `dim=1`）

我们来通过一个具体的 3x3 张量示例，观察在不同维度上使用 topk() 的结果。

import torch# 创建一个 3x3 的二维张量
x = torch.tensor([[0.1, 0.8, 0.6],[0.9, 0.2, 0.3],[0.5, 0.4, 0.7]
])

沿行取 Top-k：`dim=1`

print(torch.topk(x, k=2, dim=1))# 输出：
# values=tensor([[0.8000, 0.6000],
#         [0.9000, 0.3000],
#         [0.7000, 0.5000]]),
# indices=tensor([[1, 2],
#         [0, 2],
#         [2, 0]]))

每一行分别取出前两个最大值及其列索引

沿列取 Top-k：`dim=0`

print(torch.topk(x, k=2, dim=0))#  输出：
# values=tensor([[0.9000, 0.8000, 0.7000],
#        [0.5000, 0.4000, 0.6000]]),
# indices=tensor([[1, 0, 2],
#        [2, 2, 0]]))

每一列分别取出前两个最大值及其对应的“行号”。

理解维度的直觉图示

dim=1：按行取 top-k（对每一行，从左往右选 k 个最大值）
dim=0：按列取 top-k（对每一列，从上往下选 k 个最大值）

操作	意图	方向
`topk(x, k, dim=1)`	每行选前 k 个最大	⟶ 横向
`topk(x, k, dim=0)`	每列选前 k 个最大	⬇ 纵向

topk 与largest、sorted操作的组合

1. 取最小值：largest=False

d = torch.tensor([5, 3, 8, 1, 2])
smallest, indices = torch.topk(d, k=2, largest=False)
print("前2小的值:", smallest) 
# 输出: tensor([1, 2])

2. 不排序：sorted=False

e = torch.tensor([3, 1, 4, 2, 5])
values, indices = torch.topk(e, k=3, sorted=False)
print("前3大的值（未排序）:", values)  
# 输出: tensor([3, 4, 5])
print("对应索引:", indices)         
# 输出: tensor([0, 2, 4])

PyTorch topk() 用法详解：取最大值

torch.topk(input, k) 返回张量中最大的 k 个元素以及它们在原张量中的索引。函数原型 torch.topk(input, k, dimNone, largestTrue, sortedTrue)参数说明： 参数说明input输入张量k要取出的前 k 个值dim指定沿哪个维度取值（默认是最后一维&#xff…...

编程日记 2025/7/3 18:33:03

毕业论文查重原理及降重方法

【30%的重复率有那么重要吗？】老师说论文重读率必须低于30%，否则无法毕业！ 如果您在专科或者普通本科学生，我只能这样一句话告诉你：你想多了~， 真的想多了~~~，一篇论文还不至于让你不能毕…...

编程日记 2025/7/3 5:15:29

Golang Kratos 系列：业务分层的若干思考（二）

上一篇文章简单讨论了领域层在Kratos中的使用，主要涉及引入领域层，将数据层和业务层之间的解耦，接下来讨论一个稍微全面一点的例子，在此基础上引入外部Api（主要是易变部分）的领域层下的情况。我们同样可以…...

编程日记 2025/7/3 14:24:40

技术伦理之争：OpenAI陷抄袭风波，法院强制下架宣传视频

在AI巨头OpenAI宣布以65亿美元天价收购苹果前设计总监Jony Ive的硬件公司IO仅一个月后，一场抄袭指控将这家科技明星企业推上风口浪尖。源自谷歌X实验室的初创企业IYO将OpenAI告上法庭，指控其窃取智能耳塞核心技术，并通过巨额收购试图掩盖抄袭…...

编程日记 2025/7/1 12:36:35

烟花爆竹生产企业库房存储安全风险预警系统

烟花爆竹生产企业库房存储安全风险预警系统是保障库房物资安全、规范作业流程、防范安全事故的重要技术手段，涵盖多个关键预警功能。温湿度预警在库房内安装温湿度传感器，这些传感器如同敏锐的“环境感知员”，能够实时监测库房内环境变…...

编程日记 2025/7/1 11:20:10

Jenkins+Jmeter+Ant接口持续集成

2025最新Jmeter接口测试从入门到精通（全套项目实战教程） 前言： 为什么要用Jmeter做接口测试： 当选择这套方案的时候，很多人会问，为什么选择Jmeter做Case管理？为什么不自己写框架？说…...

编程日记 2025/7/1 12:36:33

基于STM32的寻迹小车设计

标题:基于STM32的寻迹小车设计内容:1.摘要本文围绕基于STM32的寻迹小车设计展开。背景是随着自动化技术的发展，寻迹小车在工业巡检、物流运输等领域有广泛应用前景。目的是设计一款能稳定、准确寻迹的小车。方法上，以STM32微控制器为核心，…...

编程日记 2025/7/1 12:36:32

【150】基于SSM+Vue实现的小说阅读小程序（有文档）

系统介绍基于SSMVue实现的小说阅读小程序采用前后端分离的架构方式，系统设计了管理员、用户两种角色，系统分为管理端、小程序端，管理端实现了管理员登录、个人中心、管理员管理、帮助中心管理、基础数据管理、论坛中心管理、公告资讯管理、…...

编程日记 2025/7/3 8:09:57

ValKey中使用SIMD指令优化bitcount命令

一、AVX/AVX2 的历史演进随着计算机技术的飞速发展，数据处理需求呈指数级增长，SIMD（单指令多数据）技术应运而生。它通过一条指令同时处理多个数据元素，大幅提升计算效率，从早期的 MMX 技术起步&#xff0…...

编程日记 2025/7/1 12:36:27

leetcode114-二叉树展开为链表

leetcode 114 思路用简单例子推导规律不要一开始就看复杂的树，先从最简单的情况入手案例一：只有一个节点输入：1 输出：1不需要任何操作，直接返回案例二：有两个节点输入： 1/2输出&a…...

编程日记 2025/7/1 12:36:25

1.给出下面表达式的逆波兰表示(后缀式): 3请将表达式-(ab)* (cd)-(abc)分别表示成三元式,间接三元式和四元式序列四元式(Op,arg1,arg2,result) (,a,b,T1) (,c,d,T2) (*,T1,T2,T3) (uminus,T3,-,T4) (,a,b,T5) (,T5,c,T6) (-,T4,T6,r) 三元式 (op,arg1,arg2) (0) (,…...

编程日记 2025/7/1 12:36:24

Spring Ai Alibaba Graph实现五大工作流模式

Spring Ai Alibaba Graph实现五大工作流模式概述在 building-effective-agents 一文中,Anthropic将"智能体系统"(agentic systems),从架构层面分为 “工作流”（workflows）和 “智能体”（agents）： 工作流…...

编程日记 2025/7/1 12:36:22

基于单片机的语音控制设计（论文）

摘要自然语音作为人机交互在目前得以广泛的应用以及极大的发展前景。该设计介绍了基于非指定人语音芯片LD3320的语音控制器结构及其实现语音控制的方法。该语音控制器利用STM32F103C8T6单片机作为主要控制器，控制芯片对输入的进行语音识别并处理，根据语…...

编程日记 2025/7/1 10:29:54

【网络安全】从IP头部看网络通信：IPv4、IPv6与抓包工具 Wireshark 实战

从IP头部看网络通信：IPv4、IPv6与抓包工具 Wireshark实战在网络安全分析和数据通信的世界中，一切都始于“数据包”。数据包是网络上传输的基本单位，而数据包的结构与内容，正是我们理解网络行为的核心。本文将带你深入了解 IP 协…...

编程日记 2025/7/1 12:36:16

计算机组成原理笔记（公众号版本）

1.MAR，存储单元和PC位数之间的关系我们的MAR位数取决于我们的存储单元的个数； PC位数也是取决于我们的存储单元的个数； 假设我们是64个存储单元，我们的这个MAR实际上就是6位，因为这个2的6次方等于我们的64吗&#…...

编程日记 2025/7/1 5:33:22

CPM基本原理

CPM（连续相位调制，Continuous Phase Modulation ）是一种恒包络，信号相位随时间连续变化的调制技术。其优势显著，连续相位特性，频谱效率高，对带外辐射抑制好；相位平滑过渡&#xff0…...

编程日记 2025/7/4 2:38:49

浅谈开源在线客服系统与 APP 集成的技术方案与优劣势

在为移动端 App 接入在线客服系统的过程中，我经历了长时间的技术选型探索。最初，我也曾被一些“技术理想主义”选项所吸引，比如让用户自己研发界面我提供 API 以获得最高自由度，或集成 SDK 以追求原生体验。然而，随着项…...

编程日记 2025/7/1 5:33:18

AutoGPT，自主完成复杂任务

AutoGPT是一个开源的AI Agent项目，它的核心目标是让AI能够自主完成复杂任务，而不仅仅是回答单个问题。简单来说，它让AI具备了"自主思考和行动"的能力。 1. AutoGPT的核心概念什么是AI Agent？ AI Agent（智…...

编程日记 2025/7/1 5:33:17

基于Qt C++的影像重采样批处理工具设计与实现

摘要本文介绍了一种基于Qt C++框架开发的高效影像重采样批处理工具。该工具支持按分辨率(DPI) 和按缩放倍率两种重采样模式，提供多种插值算法选择，具备强大的批量处理能力和直观的用户界面。工具实现了影像处理的自动化流程，显著提高了图像处理效率，特别适用于遥感影像处…...

编程日记 2025/7/1 5:33:15

Qt Windows平台调用ffmpeg动态库

本文基于QT6.8实测验证。一、下载预编译库‌ 从官方或第三方源获取FFmpeg的Windows动态库（Shared版本），解压后需包含以下目录： bin DLL文件（运行时依赖）include 头文件lib .lib或.dll.a链接…...

编程日记 2025/7/1 5:33:13

猿人学js逆向比赛第一届第十三题

一、分析请求通过分析请求得知，本题目的参数为yuanrenxue_cookie，因为题目中中明说了是动态cookie，所以可以先删除这个cookie然后下脚本断点，简单分析看一下页面执行的js代码逻辑。在经过这段代码的时候很明显的可以看到这里的有…...

编程日记 2025/7/1 12:36:15

React性能优化精髓之一：频繁setState导致滚动卡顿的解决方案

在开发一个 List 页面时，我们遇到了一个典型的React性能问题：页面在滚动时出现明显卡顿。这个问题的调试过程充满了误判和重新思考，最终发现了一个重要的性能优化原则。问题现象我们有一个监控仪表盘页面，包含多个图表组件。用…...

编程日记 2025/7/1 12:36:13

JavaScript 事件常用属性

一、事件对象基础在事件处理函数中，浏览器会自动传入一个 event 对象，它包含了与事件相关的所有信息。 element.addEventListener(click, function(event) {// event 是事件对象 }); 二、常见事件属性属性名类型描述 type String 事件类…...

编程日记 2025/7/1 12:36:11

配置自己的NTP 服务器做时间同步

✅ 推荐方案：使用 chrony 搭建 NTP 服务器（适用于 CentOS 7/8/9） chrony 是 CentOS 推荐的 NTP 实现，精度高、资源占用低、同步快，默认在 CentOS 8 中取代了 ntpd。 🔧 一、安装 chrony sudo yum install…...

编程日记 2025/7/3 17:49:23

基于深度学习的双色球智能预测系统：从原理到实现

需要源码的小伙伴可以在这里直接下载：基于深度学习的双色球智能预测系统：从原理到实现（完整代码训练数据）可直接运行-预测）资源-CSDN下载可直接运行，包括完整的训练测试数据，让你的双色球更准&a…...

编程日记 2025/7/3 20:10:20

STM32[笔记]--4.嵌入式硬件基础

4.嵌入式硬件基础 4.1认识上官二号开发板主控芯片：STM32F103C8T6高速晶振：8M低速晶振：32.768kLED：5颗KEY：3个主控芯片内部的资源如下项目介绍内核Cortex-M3Flsah64K*8bitSRAM20K*8bitGPIO37个GPIO，分别为PA0-PB15,PC13-PC15,PD0-PD1ADC2个12bitADC合计12了通道，外部通…...

编程日记 2025/7/1 13:50:50

Springboot项目中使用手机号短信验证码注册登录实现

文章目录 1. 功能概述2. 技术栈3. 实现步骤3.1 短信服务集成3.2 创建短信工具类3.3 验证码生成和存储3.4 控制器实现发送短信验证码手机号+验证码注册短信验证码登录3.5 服务层实现4. 前端实现4.1 API层4.2 手机号注册页面5. 最佳实践6. 总结本文将介绍如何在Spring Boot应用中…...

编程日记 2025/7/1 15:02:54

Python的GUI库选择指南（深度拓展）

前文我们分析了python的GUI库，有很多，面向应用场景也不尽相同，如何在使用过程中，选择合适的GUI库呢？可以查看：python有哪些常用的GUI（图形用户界面）库及选择指南-CSDN博客初学者推…...

编程日记 2025/7/1 12:36:07

Kubernetes生命周期管理：深入理解 Pod 生命周期

云原生学习路线导航页（持续更新中） kubernetes学习系列快捷链接 Kubernetes架构原则和对象设计（一）Kubernetes架构原则和对象设计（二）Kubernetes架构原则和对象设计（三）Kubernetes控…...

编程日记 2025/7/1 15:02:52

JVM(12)——详解G1垃圾回收器

G1（Garbage-First）垃圾回收器。它是现代 Java 应用中默认的垃圾回收器（自 JDK 9 起），旨在提供一个高性能、可预测停顿时间（低延迟）的解决方案，尤其适合大内存（多GB甚至TB…...

编程日记 2025/7/1 15:02:50

Matplotlib vs Seaborn：选择与区别

相同点都是Python数据可视化库：两者都用于创建统计图形和图表基于Python生态系统：都与NumPy、Pandas等科学计算库良好集成开源免费：两者都是开源项目，可自由使用支持多种图表类型：都能创建折线图、柱状图、散点…...

编程日记 2025/7/1 15:02:48

TCP/UDP协议深度解析（一）：UDP特性与TCP确认应答以及重传机制

🔍 开发者资源导航 🔍🏷️ 博客主页： 个人主页📚 专栏订阅： JavaEE全栈专栏前言在网络通信的世界里，传输层协议如同交通规则，决定了数据包如何从源头抵达目的地。其中UDP和TCP就…...

编程日记 2025/7/1 15:02:46

Linux线程概念及常用接口（1）

目录 1. Linux线程概念什么是线程线程的优点线程的缺点线程异常线程用途 2. Linux进程VS线程进程和线程关于进程线程的问题 3. Linux线程控制 POSIX线程库创建线程线程ID及进程地址空间布局线程终止线程等待为什么需要线程等待？ 4. 分离线程 1. Linux线…...

编程日记 2025/7/1 15:02:44

Qt+OPC开发笔记（三）：OPC客户端订阅特点消息的Demo

若该文为原创文章，转载请注明原文出处本文章博客地址：https://hpzwl.blog.csdn.net/article/details/148868209 长沙红胖子Qt（长沙创微智科）博文大全：开发技术集合（包含Qt实用技术、树莓派、三维、OpenCV…...

编程日记 2025/7/1 15:02:42

《美化生活》投稿简介

《美化生活》杂志是国家新闻出版署批准的正规期刊，是面向全国发行的文学艺术刊。本刊坚持“传播新理念，交流新经验”办刊方针，坚持“指导消费、丰富生活、美化心灵、雅俗共赏”的宗旨，以其丰富的内容，融学术性与技术性…...

编程日记 2025/7/1 15:02:40

如何为虚拟机上的 Manjaro Linux启用 VMware 拖放功能

如果你的Manjaro 发行版本是安装在 VMware Workstation Player 上使用的 ，而且希望可以通过拖放功能将文件或文件夹从宿主机复制到客户端的Manjaro 里面，那么可以按照以下的步骤进行操作，开启拖放功能。在 VMware 虚拟机上安装 Manjaro 后&…...

编程日记 2025/7/1 15:02:38

VIVADO导出仿真数据到MATLAB中进行分析

VIVADO导出仿真数据到MATLAB中进行分析目录前言一、导出仿真数据需要编写的RTL代码二、MATLAB读入txt文件中的数据三、需要注意的点总结前言在使用 Xilinx Vivado 进行 FPGA 开发时，如何将 RTL 仿真生成的数据导出，进行进一步分析与可视化&…...

编程日记 2025/7/1 15:02:37

Harmony状态管理@Event

ArkUI Event装饰器：实现子组件向父组件通信的规范方式概述 Event装饰器是ArkUI框架中用于规范组件间通信的重要工具，特别是在需要子组件向父组件请求更新Param变量的场景下。它通过回调机制实现数据的双向同步，是组件化开发中不可或缺的一…...

编程日记 2025/7/1 16:18:24

算力服务器选型

算力服务器选型需结合应用场景、算力需求及扩展性，核心要素如下： 应用定位： AI 训练 / 推理：优先高算力 GPU（如 NVIDIA A100、H100，或 RTX4090），搭配多核 CPU（如 Intel …...

编程日记 2025/7/4 1:21:32

基于目标驱动的分布式敏捷开发

研究结论风险对项目目标的影响时间目标：需求管理不当（如需求优先级不明确、多产品负责人需求冲突）、架构变更导致的返工、跨站点协调问题（如第三方依赖、通信基础设施不足）是影响项目时间的主要风险因素。质量目标&…...

编程日记 2025/7/1 9:57:52

大数据在UI前端的应用拓展：用户行为分析的深度挖掘

hello宝子们...我们是艾斯视觉擅长ui设计、前端开发、数字孪生、大数据、三维建模、三维动画10年经验!希望我的分享能帮助到您!如需帮助可以评论关注私信我们一起探讨!致敬感谢感恩! 在当今数字化时代，大数据已成为推动各行业发展的核心动力之一。对于 UI 前端而言&…...

编程日记 2025/7/4 2:20:13

elk+filebeat收集springboot项目日志

目录步骤 1: 安装和配置Elasticsearch 步骤 2: 安装和配置Logstash（可选） 步骤 3: 安装和配置Filebeat 步骤 4: 安装和配置Kibana 要使用ELK（Elasticsearch, Logstash, Kibana）堆栈和Filebeat来收集Spring Boot项目的日志&am…...

编程日记 2025/6/30 18:45:57

华为云Flexus+DeepSeek征文 | 华为云MaaS平台上的智能客服Agent开发：多渠道融合应用案例

华为云FlexusDeepSeek征文 | 华为云MaaS平台上的智能客服Agent开发：多渠道融合应用案例 🌟 嗨，我是IRpickstars！ 🌌 总有一行代码，能点亮万千星辰。 🔍 在技术的宇宙中，我愿做永不…...

编程日记 2025/6/30 11:23:49

SQL关键字三分钟入门：UPDATE —— 修改数据

在数据库操作中，除了添加新记录外，我们还需要经常修改已有的记录。例如： 更新用户的个人信息；调整订单的状态；更改产品的价格。这时候就需要用到 SQL 中非常基础但极其重要的关键字 —— UPDATE！ 它是用…...

编程日记 2025/7/1 2:44:51

用Rust写平衡三进制加法器

1、三进制加法器的发展最初的平衡三进制加法是采用了三选一结构(github原文)，这位大佬也很厉害，他是硬件都弄了出来的，也写了虚拟机，甚至用这三态多路复用器弄出了可以存状态的硬件，但我没有去看了，因为当…...

编程日记 2025/7/3 20:10:22

【AI时代速通QT】第三节：Linux环境中安装QT并做测试调试

目录引言一、Linux QT开发环境的核心要素 1.1 编译器（g）与构建工具（make） 1.2 搞定 OpenGL 依赖二、核心步骤——安装 Qt Creator 2.1 获取官方在线安装器 2.2 赋予文件执行权限 2.3 运行图形化安装向导三、Linux上创…...

编程日记 2025/6/30 21:29:30

论文阅读：2025 arxiv Qwen3 Technical Report

https://arxiv.org/pdf/2505.09388 https://www.doubao.com/chat/9918384373236738 文章目录论文翻译Qwen3技术报告摘要1 引言论文翻译 Qwen3技术报告 Qwen团队摘要在这项工作中，我们介绍了Qwen模型家族的最新版本Qwen3。Qwen3包含一系列大型语言模型&…...

编程日记 2025/6/30 23:39:09