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