mindspore.ops.argwhere

mindspore.ops.argwhere(input)

返回一个tensor，包含输入tensor所有非零元素的位置。

参数：

• input (Tensor) - 输入tensor。

返回：

二维tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([[[1,  0], [-5, 0]]], mindspore.int32)
>>> output = mindspore.ops.argwhere(x)
>>> print(output)
[[0 0 0]
 [0 1 0]]