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

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

相关推荐
wanghao6664553 分钟前
机器学习三大流派:监督、无监督与强化学习
人工智能·机器学习
爱喝可乐的老王5 分钟前
神经网络的基础:核心是 “搭积木 + 激活信号”
人工智能·深度学习·神经网络
梁辰兴13 分钟前
FSD入华将如何改变我国自动驾驶市场格局?
人工智能·科技·机器学习·自动驾驶·特斯拉·fds·梁辰兴
AI营销实验室20 分钟前
AI营销破解券商获客难引领2026增长新范式
人工智能·microsoft
njsgcs25 分钟前
ppo可以不需要提取特征,直接训练ac吗。ppo不知道自己现在在第几步吗
人工智能·ppo
lixin5565561 小时前
基于深度生成对抗网络的高质量图像生成模型研究与实现
java·人工智能·pytorch·python·深度学习·语言模型
泰迪智能科技011 小时前
泰迪智能科技人工智能综合实验箱功能简介及实训支持内容介绍
人工智能·科技
DS随心转小程序1 小时前
DeepSeek井号解决方法
人工智能·aigc·deepseek·ds随心转
安全二次方security²1 小时前
CUDA C++编程指南(7.15&16)——C++语言扩展之内存空间谓词和转化函数
c++·人工智能·nvidia·cuda·内存空间谓词函数·内存空间转化函数·address space
laplace01231 小时前
大模型整个训练流程
人工智能·深度学习·embedding·agent·rag