mindspore.ops.Bincount

class mindspore.ops.Bincount[源代码]

计算整数数组中每个值的出现次数。

警告

这是一个实验性API,后续可能修改或删除。

输入:
  • array (Tensor) - int32数据类型的Tensor。

  • size (Tensor) - int32数据类型的非负Tensor。

  • weights (Tensor) - 与 array shape相同或维度为0的Tensor。当维度为0时,所有权重均等于1。数据类型必须是以下类型之一:int32、int64、float32、float64。

输出:

Tensor,数据类型与 weights 相同。

异常:
  • TypeError - array 的数据类型不是int32。

  • TypeError - size 的数据类型不是int32。

  • ValueError - size 中的数据存在负值。

  • ValueError - weights 是空Tensor。

  • ValueError - weights 的size不为0且和 array 的shape不同。

  • TypeError - weights 的数据类型不是int32、int64、float32或float64。

支持平台:

Ascend GPU CPU

样例:

>>> array = Tensor(np.array([1, 2, 2, 3, 3, 3, 4, 4, 4, 4]), mindspore.int32)
>>> size = Tensor(5, mindspore.int32)
>>> weights = Tensor(np.array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1]), mindspore.float32)
>>> bincount = ops.Bincount()
>>> bins = bincount(array, size, weights)
>>> print(bins)
[0. 1. 2. 3. 4.]