PyTorch 统计属性-Tensor基本操作

原创

已于 2024-06-13 10:08:58 修改 · 974 阅读

17 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #人工智能 #python

于 2024-06-12 09:25:45 首次发布

最小 min, 最大 max, 均值 mean，累加 sum，累乘 prod …

>>> a = torch.arange(0,8).view(2,4).float()
>>> a
tensor([[0., 1., 2., 3.],
        [4., 5., 6., 7.]])
        
        
>>> a.min()  	## 最小值：tensor(0.)
>>> a.max()  	## 最大值：tensor(7.)

>>> a.argmin()	## 最小值对应的 idx: tensor(0)
>>> a.argmax()	## 最大值对应的 idx: tensor(7)

>>> a.argmin(dim=1)