0
点赞
收藏
分享

微信扫一扫

pytorch学习笔记(十)——统计属性

北邮郭大宝 2022-03-21 阅读 100
java后端

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

举报

相关推荐

0 条评论