mindspore.ops.argwhere

mindspore.ops.argwhere(input)

Return a tensor of containing the positions of all non-zero elements in the input tensor.

Parameters

input (Tensor) – The input tensor.

Returns

2-D Tensor

Supported Platforms:

Ascend GPU CPU

Examples

>>> 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]])