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

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

相关推荐
数据智能老司机2 小时前
PyTorch 深度学习——使用神经网络来拟合数据
pytorch·深度学习
数据智能老司机2 小时前
PyTorch 深度学习——用于图像的扩散模型
pytorch·深度学习
数据智能老司机2 小时前
PyTorch 深度学习——Transformer 是如何工作的
pytorch·深度学习
闲云一鹤3 小时前
本地部署 B 站 IndexTTS2 模型 - AI 文本生语音神器
前端·人工智能
前端双越老师3 小时前
Skills 是什么?如何用于 Agent 开发?
人工智能·node.js·agent
yiyu071613 小时前
3分钟搞懂深度学习AI:环境安装与工具使用
人工智能·深度学习
冬奇Lab15 小时前
一天一个开源项目(第44篇):GitNexus - 零服务器的代码智能引擎,为 AI Agent 构建代码库知识图谱
人工智能·开源·资讯
冬奇Lab15 小时前
OpenClaw 深度解析(七):安全模型与沙盒
人工智能·开源
IT_陈寒17 小时前
别再死记硬背Python语法了!这5个思维模式让你代码量减半
前端·人工智能·后端
Ray Liang17 小时前
彻底治愈AI“失忆”和胡说八道的真正办法
人工智能·rag·智能体·ai助手·mindx