【pytorch10】统计属性

常见统计属性

• 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