pytorch中一些常用函数（1）

最新推荐文章于 2024-05-30 19:10:16 发布

原创

最新推荐文章于 2024-05-30 19:10:16 发布 · 2.7k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#计算机视觉

pytorch中的数学函数

在这里插入图片描述

torch.histc（）：指定最小值最大值和bin数，对tensor中的数据进行区段统计。

a = torch.rand(2,2)*10
print(a)
#6表示取6个区间段，0，0，表示采用Tensor中的最大值和最小值，也可以自己更改区间上的最大值和最小值
print( torch.histc(a,6, 0, 0) )
运行结果如下：
tensor([[5.5695, 1.2099],
        [1.3061, 1.0443]])   #a的值
tensor([3., 0., 0., 0., 0., 1.]) #6个区间段中数据的的统计

注：torch.bincount()只支持一维的Tensor

a = torch.randint(0, 10, [10]) #取0-10中间任意10个数
print(a)
print( torch.bincount(a) )
运行结果如下：
tensor([8, 1, 2, 9, 9, 8, 1, 7, 8, 7])  #a的取值
tensor([0, 2, 1, 0, 0,