mindspore.Tensor.nonzero

mindspore.Tensor.nonzero()[源代码]

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

返回:

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

支持平台:

GPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[[1,  0], [-5, 0]]]), mindspore.int32)
>>> output = x.nonzero()
>>> print(output)
[[0 0 0]
 [0 1 0]]