torch.histc

torch.histc(input, bins=100, min=0, max=0, *, out=None) → Tensor

计算张量的直方图。

元素被分入 min 和 max 之间等宽的箱子（bins）中。如果 min 和 max 均为零，则使用数据中的最小值和最大值。

小于 min、大于 max 的元素以及 NaN 元素会被忽略。

参数:

• input (Tensor) – 输入张量。

• bins (int) – 直方图的箱数

• min (Scalar) – 范围的下界（包含）

• max (Scalar) – 范围的上界（包含）

关键字参数:

out (Tensor, optional) – 输出张量。

返回:

以张量形式表示的直方图

返回类型:

张量

示例

>>> torch.histc(torch.tensor([1., 2, 1]), bins=4, min=0, max=3)
tensor([ 0.,  2.,  1.,  0.])