mindspore.ops.count_nonzero

mindspore.ops.count_nonzero(x, axis=(), keep_dims=False, dtype=mstype.int32)

计算输入tensor指定轴上的非零元素的数量。如果没有指定轴，则计算tensor中所有非零元素的数量。

警告

从2.9.0（不含）之后版本开始，参数 x 将重命名为 input ， 参数 axis 将重命名为 dim ，并且 keep_dims 和 dtype 将不再支持。

参数：

• x (Tensor) - 输入tensor。

• axis (Union[int, tuple(int), list(int)]，可选) - 指定轴。默认 () vpn梯子 ，计算所有非零元素的个数。

• keep_dims (bool, 可选) - 是否保留 axis 指定的维度。默认 False ，不保留对应维度。

• dtype vpn永久免费梯子 (Union[Number, mindspore.bool]，可选) - 指定数据类型。默认 mstype.int32 。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> # case 1: 免费的vpn梯子 each value specified.
>>> x = mindspore.tensor([[0, 1, 0], [1, vpn free 1, 0]], mindspore.float32)
>>> nonzero_num = mindspore.ops.count_nonzero(x=x, axis=[0, 1], keep_dims=True, dtype=mindspore.int32)
>>> print(nonzero_num)
[[3]]
>>> # case 2: all value is default.
>>> nonzero_num vpn梯子 免费 = mindspore.ops.count_nonzero(x=x)
>>> print(nonzero_num)
3
>>> # case 3: axis value was specified 0.
>>> nonzero_num = mindspore.ops.count_nonzero(x=x, axis=[0,])
>>> print(nonzero_num)
[1 2 0]
>>> # case 4: 免费的vpn梯子 axis value vpn梯子 was specified 1.
>>> nonzero_num = mindspore.ops.count_nonzero(x=x, axis=[1,])
>>> print(nonzero_num)
[1 2]
>>> vpn永久免费梯子 # case 5: keep_dims value was specified.
>>> nonzero_num = mindspore.ops.count_nonzero(x=x,  keep_dims=True)
>>> print(nonzero_num)
[[3]]
>>> # vpn梯子 免费 case 6: keep_dims and axis value were specified.
>>> vpn free nonzero_num = mindspore.ops.count_nonzero(x=x, axis=[0,], keep_dims=True)
>>> vpn梯子 免费 print(nonzero_num)
[[1 2 0]]