mindspore.mint.count_nonzero
============================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/mint/mindspore.mint.func_count_nonzero.rst
    :alt: 查看源文件


.. py:function:: mindspore.mint.count_nonzero(input, dim=None)

    在给定维度 `dim` 上，计算 Tensor `input` 中非零元素的数量。如果未指定维度，则计算Tensor中所有非零元素的数量。

    .. warning::
        这是一个实验性API，后续可能修改或删除。

    参数：
        - **input** (Tensor) - 计算非零元素个数的输入。其shape为 :math:`(*)` ，其中 :math:`*` 为任意维度。
        - **dim** (Union[None, int, tuple(int), list(int)], 可选) - 计算非零元素数量的维度。默认值： ``None`` ，表示计算所有非零元素的个数。如果 `dim` 为 ``None`` ，则对Tensor中的所有元素求和。

    返回：
        Tensor，指定的轴上非零元素数量。

    异常：
        - **TypeError** - `input` 不是Tensor类型。
        - **TypeError** - `dim` 类型不是int、tuple(int)、list(int)或None。
        - **ValueError** - `dim` 取值不在 :math:`[-input.ndim, input.ndim)` 范围。