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]]