mindspore.ops.HistogramFixedWidth

查看源文件
class mindspore.ops.HistogramFixedWidth(nbins, dtype='int32')[源代码]

返回一个rank为1的直方图,该直方图中的每个组的值表示数量。每个组的宽度应该相等,且宽度由输入 range 和参数 nbins 决定。

参数:
  • nbins (int) - 直方图的组数,类型为正整数。

  • dtype (str, 可选) - 可选属性。数据类型必须为str。默认值: 'int32'

输入:
  • x (Tensor) - HistogramFixedWidth的输入,为一个Tensor。数据类型必须为int32、float32、float16或float64。

  • range (Tensor) - 数据类型与 x 相同,shape为 \((2,)\) 。x <= range[0] 将映射到histogram[0],x >= range[1]将映射到histogram[-1]。

输出:

1-D Tensor,其长度为 nbins ,数据类型为int32。

异常:
  • TypeError - 如果 dtype 不是str或 nbins 不是int。

  • ValueError - 如果 nbins 小于1。

  • ValueError - 如果 dtype 不是 “int32”。

支持平台:

Ascend GPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> x = Tensor([-1.0, 0.0, 1.5, 2.0, 5.0, 15], mindspore.float16)
>>> range_op = Tensor([0.0, 5.0], mindspore.float16)
>>> hist = ops.HistogramFixedWidth(5)
>>> output = hist(x, range_op)
>>> print(output)
[2 1 1 0 2]