mindspore.ops.NonZero

class mindspore.ops.NonZero

计算输入Tensor中所有非零元素的索引位置。

更多细节请参考 mindspore.ops.nonzero()。

输入：

• x (Tensor) - 输入Tensor，其秩应大于等于1。

输出：

• y (Tensor) - 二维Tensor，数据类型为int64。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore.ops import NonZero
>>> x = Tensor(np.array([[[1,  0], [-5, 0]]]), mindspore.int32)
>>> nonzero = NonZero()
>>> output = nonzero(x)
>>> print(output)
[[0 0 0]
 [0 1 0]]
>>> x = Tensor(np.array([1, 0, 2, 0, 3]), mindspore.int32)
>>> nonzero = NonZero()
>>> output = nonzero(x)
>>> print(output)
[[0]
 [2]
 [4]]