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列。

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

相关推荐
kTR2hD1qb6 小时前
近期使用Claude Code + Opus4.7设计开发了一个开源项目:Qianyuan AI Agentic Framework
人工智能·开源
老兵发新帖6 小时前
ECC开源项目分析
人工智能
寻道码路6 小时前
LangChain4j Java AI 应用开发实战(十):Embedding 模型与文本分类 - 语义向量化
java·人工智能·ai·embedding
星夜夏空996 小时前
FreeRTOS学习(6)——任务创建
单片机·嵌入式硬件·学习
春生野草6 小时前
大模型--mcp、skill和工作流
人工智能
John_ToDebug6 小时前
Skills 系统深度解析:概念、定位与加载时机
人工智能·经验分享·ai
weixin_468466856 小时前
图像滤波算法新手实战指南
图像处理·人工智能·算法·计算机视觉·ai·机器视觉·滤波
程序大视界6 小时前
AI重塑教育:2026年教育行业AI应用全景报告与技术解析
人工智能·教育
麦哲思科技任甲林6 小时前
白话skills之三:Skills与程序的区别
人工智能·编排·skills
nashane6 小时前
HarmonyOS 6学习:保存图片预览空白?沙箱路径转URI的“视觉修复”术
学习·华为·harmonyos