mindspore.ops.NonZero
- class mindspore.ops.NonZero[源代码]
计算输入Tensor中所有非零元素的索引位置。
更多细节请参考
mindspore.ops.nonzero()
。- 输入:
x (Tensor) - 输入Tensor。
- 输出:
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]]