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

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

相关推荐
KaMeidebaby23 分钟前
卡梅德生物技术快报|纯化重组蛋白实操详解
人工智能·python·tcp/ip·算法·机器学习
Cloud_Shy61825 分钟前
解读《Effective Python 3rd Edition》:从练气到老魔(第五章 Item 30 - 32)
开发语言·人工智能·笔记·python·学习方法
YueTann27 分钟前
OpenRLHF设计
人工智能
云烟成雨TD29 分钟前
Spring AI 1.x 系列【52】可观测集成 SkyWalking
人工智能·spring·skywalking
云烟成雨TD29 分钟前
Spring AI 1.x 系列【57】动态工具发现:Tool Search Tool
java·人工智能·spring
AndrewHZ30 分钟前
【LLM技术全景】规模定律与模型演进:为什么模型越大越强?
人工智能·gpt·深度学习·语言模型·llm·openai·规模定律
galaxylove30 分钟前
Gartner发布创新洞察:AI SOC智能体加速通信运营商安全运营转型
大数据·人工智能·安全
甩手网软件40 分钟前
Shopee2026新规:费率重构与履约收紧下,卖家如何破局?
大数据·人工智能
数据库小学妹41 分钟前
AI时代数据库怎么选?多模融合、数据统一存储与选型实战指南
数据库·人工智能·经验分享·ai
lizhihai_991 小时前
股市学习心得-AI 产业链核心标的梳理清单
大数据·服务器·人工智能·科技·学习