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

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

相关推荐
zhangfeng11336 小时前
国家超算中心 scnet.cn 跨用户文件分享流程总结 多个用户之间 文件共享 不需要反复下载上传
人工智能·语言模型·大模型
ting94520009 小时前
Tornado 全栈技术深度指南:从原理到实战
人工智能·python·架构·tornado
果汁华9 小时前
Browserbase Skills:让 Claude Agent 真正“看见“网页世界
人工智能·python
ZhengEnCi9 小时前
04-缩放点积注意力代码实现 💻
人工智能·python
HackTwoHub10 小时前
AI大模型网关存在SQL注入、附 POC 复现、影响版本LiteLLM 1.81.16~1.83.7(CVE-2026-42208)
数据库·人工智能·sql·网络安全·系统安全·网络攻击模型·安全架构
段一凡-华北理工大学10 小时前
【高炉炼铁领域炉温监测、预警、调控智能体设计与应用】~系列文章08:多模态数据融合:让数据更聪明
人工智能·python·高炉炼铁·ai赋能·工业智能体·高炉炉温
网络工程小王11 小时前
【LangChain 大模型6大调用指南】调用大模型篇
linux·运维·服务器·人工智能·学习
HIT_Weston11 小时前
63、【Agent】【OpenCode】用户对话提示词(示例)
人工智能·agent·opencode
qq_5710993511 小时前
学习周报四十三
学习
CV-杨帆11 小时前
Phi-4-mini-flash-reasoning 部署安装与推理测试完整记录
人工智能