设计
规格
API
API映射
迁移指南
FAQ
RELEASE NOTES
计算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]]