统计属性
<
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 等价于(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基本运算
本文发布于:2024-01-28 13:16:57,感谢您对本站的认可!
本文链接:https://www.4u4v.net/it/17064190207689.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
留言与评论(共有 0 条评论) |