mindspore.ops.NonZero
- class mindspore.ops.NonZero[source]
Return a Tensor of the positions of all non-zero values.
- Inputs:
input (Tensor) - The input Tensor, its rank should be greater than or eaqual to 1.
- Outputs:
Tensor, a 2-D Tensor whose data type is int64, containing the positions of all non-zero values of the input.
- Raises
TypeError – If input is not Tensor.
ValueError – If dim of input equals to 0.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input = Tensor(np.array([1, 0, 2, 0, 3]), mindspore.int32) >>> output = ops.NonZero()(input) >>> print(output) [[0] [2] [4]]