mindspore.numpy.count_nonzero

查看源文件
mindspore.numpy.count_nonzero(x, axis=None, keepdims=False)[源代码]

计算Tensor x 中的非零值数量。

参数：

  • x (Tensor) - 需要统计非零值数量的Tensor。

  • axis (Union[int,tuple], 可选) - 指定沿着哪些轴统计非零元素个数，可以是单个轴，也可以用tuple表示多个轴。 默认值为 None，此时沿完全展平的 x 计算非零值数量。 默认值： None 。

  • keepdims (bool, 可选) - 如果设置为 True ，结果会保留计数所沿的 axis ，该维度的大小为1。若使用此选项，结果会广播到 和 x 同一个维度数。 默认值： False 。

返回：

Tensor， 表示 x 沿给定 axis 的非零值数量。 否则，返回 x 中非零值的总数。

异常：

  • TypeError - 如果 axis 不是int或tuple。

  • ValueError - 如果 axis 不在范围[-x.ndim, x.ndim)内。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> x = np.asarray([1, 2, 3, -4, 0, 3, 2, 0])
>>> output = np.count_nonzero(x)
>>> print(output)
6