mindspore.ops.nonzero

mindspore.ops.nonzero(x)

计算x中非零元素的下标。

参数：

• x (Tensor) - nonzero的输入，任意维度的Tensor，秩应大于1。其数据类型为数值型和布尔型。

返回：

Tensor，维度为2，类型为int64，表示输入中所有非零元素的下标。

异常：

• TypeError - x 不是Tensor。

• ValueError - x 的维度为0。

支持平台：

Ascend GPU CPU

样例：

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