mindspore.ops.nonzero
- mindspore.ops.nonzero(x)[源代码]
计算x中非零元素的下标。
- 参数:
x (Tensor) - nonzero的输入,任意维度的Tensor,秩应大于1。其数据类型为数值型和布尔型。
- 返回:
Tensor,维度为2,类型为int64,表示输入中所有非零元素的下标。
- 异常:
TypeError - x 不是Tensor。
ValueError - x 的维度为0。
- 支持平台:
GPU
样例:
>>> 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]]