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

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

相关推荐
JEECG低代码平台21 分钟前
GitHub 十大 Java 语言 AI 开源项目推荐
java·人工智能·github
T***u33323 分钟前
后端缓存技术学习,Redis实战案例
redis·学习·缓存
Cathyqiii34 分钟前
传统扩散模型 VS Diffusion-TS
人工智能·算法
海边夕阳200638 分钟前
【每天一个AI小知识】:什么是逻辑回归?
人工智能·算法·逻辑回归
Gorgous—l43 分钟前
数据结构算法学习:LeetCode热题100-图论篇(岛屿数量、腐烂的橘子、课程表、实现 Trie (前缀树))
数据结构·学习·算法
im_AMBER1 小时前
算法笔记 13 BFS | 图
笔记·学习·算法·广度优先
非著名架构师1 小时前
团雾、结冰、大风——高速公路的“隐形杀手”:智慧气象预警如何为您的路网安全保驾护航
人工智能·新能源风光提高精度·疾风气象大模型4.0·疾风气象大模型·风光功率预测
IT_陈寒1 小时前
Redis深度优化:10个让你的QPS提升50%的关键配置解析
前端·人工智能·后端
2501_941142931 小时前
5G与边缘计算结合在智能物流系统中的高效调度与实时监控应用研究
人工智能
2501_941144421 小时前
边缘计算与人工智能在智能制造生产线优化与故障预测中的应用研究
人工智能·边缘计算·制造