mindspore.ops.NonZero
- class mindspore.ops.NonZero[源代码]
返回所有非零元素的索引位置。
- 输入:
input (Tensor) - 输入Tensor,其秩应大于等于1。
- 输出:
二维Tensor,数据类型为int64,包含所有输入中的非零元素的索引位置。
- 异常:
TypeError - 如果 input 不是Tensor。
ValueError - 如果 input 的维度等于0。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> 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]]