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 分钟前
SQL学习之查询
java·sql·学习
蓝星空20004 分钟前
GPT-image-2 模型国内免费使用教程:零门槛直出商业级 AI 大片
人工智能·gpt·aigc·image2·imagen
直接冲冲冲5 分钟前
两层神经网络与三层神经网络的区别
人工智能·深度学习·神经网络
从零开始的代码生活_5 分钟前
C++ stack、queue 与 priority_queue:容器适配器原理与实战
开发语言·c++·后端·学习·算法
Elastic 中国社区官方博客10 分钟前
4 个英伟达人工智能任务,1 个 Elasticsearch 接口:嵌入、聊天、completion 和重排序
大数据·数据库·人工智能·elasticsearch·搜索引擎·ai
铅笔侠_小龙虾11 分钟前
Rust 学习(6)-所有权规则、移动语义、Clone 与 Copy
python·学习·rust
Tom·Ge17 分钟前
【AI前沿】2026年7月第三周:GPT-5.6开启递归自我改进、Claude惊现“意识结构“、中国模型周调用量全球第一
人工智能·gpt
kiss strong20 分钟前
AI学习-langchain
学习·langchain
码云骑士21 分钟前
65-AI-Agent基础原理-AutoGPT-四大架构全景解读
人工智能·python·架构
染指111025 分钟前
56.llama_index-查询引擎
人工智能·llama·rag·llama_index·llamaindex