常见统计属性
- norm(范数)
- mean,sum
- prod
- max,min,argmin,argmax
- kthvalue,topk
kthvalue求第几个的位置和第几个的值
topk求top几的这样的一个数值
norm范数
这里的norm表达的是范数的意思,normalize是正则化
matrix norm 和 vector norm有区别
norm-p
1范数是所有元素的求和
2范数是所有元素绝对值的平方和再开根号
mean,sum,min,max,prod,argmin,argmax
prod 是累乘
最大值的索引返回的是7不是[1,3],意味着mean,min,max等这些函数不带参数的话,会把tensorA reshape成一个维度然后再求max的值和max所在的索引,如果不想打平,希望在某一个维度上求max等,必须给定维度
dim,keepdim
keepdim 保持原来的维度,因为统计信息包含消dimension的操作
Top-k or k-th
- topk
- kthvalue
topk 例如对于一张照片会有一个可能性[0.1,0.2,0.6,…]取概率最大的两个概率,假设0.6和0.2是最大的两个概率,我们用topk(2)会返回[0.6,0.2]和所在索引[2,1],也就是topk要比max返回更多的数据
如果要求最小的k个,就把largest设置成为False
kthvalue就是第k个小的value,第8小的值就是第3大的值
compare
eq() 返回的是每个元素是否相等,返回值是一个矩阵
equal() 返回的是矩阵是否相等,返回值是True和False