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

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

相关推荐
fanxiaohui121382 分钟前
2026符合上飞机标准的充电宝推荐南孚传应,质量稳定又安全
运维·服务器·人工智能·科技
TechEdu2026063 分钟前
[人工智能]2026人工智能与大语言模型芯片家族参考V02:华为、Nvidia、AMD、Intel、Qualcomm
人工智能·ai
卷无止境4 分钟前
从终端里长出来的 IDE,oh-my-pi 到底是个什么东西
人工智能·后端
我滴老baby6 分钟前
工业物联网数据库选型:把计算能力放回第一维度
数据库·人工智能·架构·pdf
程序员yu8 分钟前
数智码力:Python作用域完全梳理,全局变量局部变量不再混乱
开发语言·python·学习
墨林陌10 分钟前
AI 热点日报(2026-09-18):华为昇腾960超节点发布,OpenAI 首次公开模型失准报告
人工智能
RisunJan11 分钟前
AI 每日要闻总结(2026-09-17)
人工智能
小lu飞13 分钟前
带 AI 问答的小程序选型:自建工作流与零代码生成平台的计费管理对比
人工智能
牧羊人.33318 分钟前
自然语言处理基础 01|语言转换与Word2Vec
人工智能·深度学习·自然语言处理
北城笑笑19 分钟前
Python Dev 02 & Tkinter webbrowser 实战,手写第一个 Python 桌面 GUI 小工具
人工智能·python·pip