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

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

相关推荐
GISer_Jing几秒前
AI Agent 智能体系统:A2A通信与资源优化之道
人工智能·aigc
Dev7z5 分钟前
基于深度学习的车辆分类方法研究与实现-填补国内新能源车型和品牌识别空白
深度学习·yolo
进阶小白猿7 分钟前
Java技术八股学习Day20
java·开发语言·学习
yusur14 分钟前
边缘智算新引擎 DPU 驱动的算力革新
人工智能·科技·rdma·dpu
视觉&物联智能16 分钟前
【杂谈】-企业人工智能的变革与机遇
人工智能·ai·aigc·agi
图生生26 分钟前
电商主图快速修改方案:AI工具实现元素自由增删,降低开发与设计成本
人工智能
Deepoch27 分钟前
Deepoc具身模型开发板:重新定义机器人智能化的技术底座
人工智能·机器人·具身模型·deepoc
G***技30 分钟前
搭载RK3588处理器,IM1-707核心板撑起建筑机器人“精准+高效”
人工智能
lingling00932 分钟前
2026 年 BI 发展新趋势:AI 功能如何让数据分析工具 “思考” 和 “对话”?
大数据·人工智能·数据分析
鹧鸪云光伏34 分钟前
光伏项目多,如何高效管理?
大数据·人工智能·光伏