pytorch学习笔记（十）——统计属性

tensor统计属性​​​​​​​ 

1.norm-p

范数p=1：绝对值之和

范数p=2：绝对值平方之和开2次方

......

eg:

2.mean, sum, min, max, prod(累乘)

3.argmin, argmax（不给定dim参数时，打平后最大最小值的索引）

a=torch.randn(4,10) #创建一个4行10列的Tensor

不给定dim参数，求最大值所在的索引

a.argmax()  #结果为：tensor(18) 即18是最大值所在索引

给定dim=1，求每一行最大值所在索引

a.argmax(dim=1)  #结果为：tensor(3,8,6,4) 即3，8，6，4分别是第1，2，3，4行最大值所在索引

4. dim, keepdim

keepdim使统计操作后dim保持与原来一直

5.Top-k or k-th

.topk：求最大的k个的值和索引；当参数largest=False时，为求最小的k个的值和索引。

kthvalue：第k小的值和索引

6.compare

>, >=, <, <=, !=, ==

torch.gt(a,0)表示a>0

torch.eq(a, b)和torch.equal(a, b)

torch.equal(a, b)表示a与b完全相同，返回Ture/False