torch.histc

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

计算张量的直方图。

元素会被分类到minmax之间的等宽区间内。如果minmax都为零,则使用数据的最小值和最大值。

低于 min 或高于 max 的元素,以及 NaN 元素将被忽略。

参数
  • input (Tensor) – 需要输入的张量。

  • bins (int) – 指定直方图中的柱的数量

  • min (标量) – 范围的下限(包含在内)

  • max (标量) – 范围的上界(包含在内)

关键字参数

out (Tensor, 可选) – 指定输出张量。

返回值

以张量形式表示的直方图

返回类型

张量

示例:

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