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

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

相关推荐
灵机一物4 分钟前
灵机一物AI原生电商小程序、PC端(已上线)-智谱唐杰重磅发声:原生多模态模型数月内上线!2026 年 AI 主战场转向长时程任务与自主进化
人工智能
Black蜡笔小新5 分钟前
企业私有化AI训练推理一体工作站DLTM重构企业AI开发新模式,开启智能AI新时代
人工智能·重构
jay神7 分钟前
基于YOLOv8的交通标志识别Web系统
前端·人工智能·深度学习·yolo·机器学习·毕业设计
redaijufeng7 分钟前
【无标题】
大数据·人工智能
programhelp_11 分钟前
亚麻 AI Assisted Coding OA 体验 + 避坑指南
人工智能
user298769827065411 分钟前
六、MCP 基础:协议介绍与配置
人工智能
东方佑12 分钟前
深圳先创数字技术(先创数字)避雷实录:所谓的“管理”就是极限施压
人工智能
用户357540826209815 分钟前
从学生项目到生产级 AI Agent
人工智能
AI周红伟18 分钟前
Token工厂:无锡部署昇腾384超节点算力集群,制造Token
大数据·人工智能·windows·百度·copilot·制造
nap-joker23 分钟前
一种新的本体引导的属性划分集成学习模型,用于应用定量结构磁共振早期预测早产儿的认知缺陷
人工智能·机器学习·集成学习