统计属性
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基本运算



