mindspore.ops.nonzero

查看源文件
mindspore.ops.nonzero(input)[源代码]

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

参数:
  • input (Tensor) - 输入Tensor,其秩应大于等于1。

返回:

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

异常:
  • TypeError - input 不是Tensor。

  • ValueError - input 的维度为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]]
>>> x = Tensor(np.array([1, 0, 2, 0, 3]), mindspore.int32)
>>> output = ops.nonzero(x)
>>> print(output)
[[0]
 [2]
 [4]]