栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Pytorch 03天——统计属性、高阶操作

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Pytorch 03天——统计属性、高阶操作

统计属性

1.norm
1、norm是范数的意思,normalize、batch_norm是归一化
2、matrix norm 和vector norm 是有区别的




取哪个维度的范数,哪个维度将要被消掉。
2、mean(均值),sum,min,max,prod(累乘)
3、argmin,argmax
对于argmin,argmax:如果不给出固定的dimension,会把tensor打平成dim=1,然后返回最小、最大的索引。


4、dim、keepdim

希望结果的维度(dim)和a保持一致
5、kthvalue,topk
kthvalue 第几小的值
topk比max返回更多的数据

六、>,>=,<,<=,!=,==
1、a>0 返回的是对应的mask
2、a>0 等价于torch.ge(a,0)
3、torch.eq(a,b),比较a,b中的每个元素的值,返回一个mask
4、torch.equal(a,b) 比较a,b中每个值,返回True or False

高阶操作
1、where
torch.where(condition,a,b)->tensor c:c中数值的来源于:a,b

2、Gather
torch.gather(input,dim,index,out=None) -> Tensor
input:表示要查的表
dim:对input查找的维度
index:查找的索引值

借鉴
pytorch学习笔记5–pytorch基本运算

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/324411.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号