torch.max()学习记录

x=tourch.tensor(\[1,2,3,4,6,5,9,11,4,-2,6,20])为4*3的一个张量

1)请思考y的值

y=torch.max(x,dim=0)

分析:x是一个2D: 4*3的张量,dim=0 ,表示按照行操作,得到的张量shape=3,对应的是[1,4,9,-2】,【2,6,11,6】,【3,5,4,20】三组内挑选最大值组分别是9、11、20成为【9,11,20】

因此最大应该是y=tensor(9,11,20) ,对应的argmax()=tensor(2,2,3),表示【9,11,20】三个分别在第2行、第二行、第三行。

2)请思考y的值

y=torch.max(x,dim=1)

分析:x是一个2D: 4*3的张量,dim=1 ,表示按照列操作,得到的张量shape=4,我的理解是1,2,3的最大为3,4,5,6的最大为6,【9,11,4]最大值为11,,-2,6,20最大值为20,因此y=tensor(3,6,11,20),对应的argmax()=tensor(2,1,1,2),表示【3,6,11,20】中,3处于第二列,6处于第1列,11处于第1列,20处于第2列。

太容易忘记,留下记录,非常基础的知识,年龄太大,肤浅得不好意思

相关推荐
喵叔哟3 分钟前
第二周概述
人工智能·langchain
leijiwen8 分钟前
LinkLifeVerse OS :数字经济时代的产业价值操作系统(Industry Value Operating System)
人工智能·云原生·saas·paas
meilindehuzi_a9 分钟前
远程 MCP 项目实战:LangChain 连接高德地图、Chrome DevTools 与文件系统
人工智能·langchain·chrome devtools
CoreTK芯通康EMC整改14 分钟前
《2026 电子制造业 EMC 合规白皮书》解读:PCB EMC 设计全链路优化指南(附器件选型矩阵)
人工智能·硬件工程·设计规范·emc整改案例·emc整改
在世修行15 分钟前
第19篇:显微镜像素比例校准 — 测量精度的基石
人工智能·计算机视觉·像素比例
辞旧 lekkk19 分钟前
【Qt系统相关】鼠标事件
linux·开发语言·qt·学习·计算机外设·萌新
zxfeng~20 分钟前
伴眸-居家养老AI 眼镜智能体 技术方案
人工智能·agent·ai编程·智能眼镜
dozenyaoyida23 分钟前
AI与大模型新闻日报 | 2026-07-11
人工智能·ai·大模型·新闻
qcx2323 分钟前
【AI Daily】AI日报 2026-07-12
人工智能
FeelTouch Labs24 分钟前
深度学习模型架构——Transformer
人工智能·深度学习·transformer