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)
Tensor(shape=[2, 3], dtype=Int64, value=
[[0, 0, 0],
[0, 1, 0]])