Pytorch基本使用系列（六）数据统计

a.norm(p,dim) 求解范数

a.min()，max(), mean(), prod()累乘, sum()

argmax() 最大值所在的索引

argmin() 最小值所在的索引 //不给定维度会打平 需要给定维度才会有索引

 

a,max(dim=1, keepdim = True) 保持dim与原始的其他维度一致

topk 前多少个最大 前多少个最小

a.topk(num, dim = 1, largest = False) False最小/ True最大

a.kthvalue(kth, dim, keepdim) 第k小的值 

 

基本的比较操作

torch.gt(a,0)

torch.eq(a,b) 查看元素是否相等

 

一些高级操作：

torch.where(condition,x,y) if  condition is true, xi, else yi //避免for循环使用CPU而没有使用GPU