mindspore.ops.argwhere

mindspore.ops.argwhere(input)

返回一个Tensor，包含所有输入Tensor非零数值的位置。

参数：

• input (Tensor) - 输入Tensor。类型可以为Number或bool。

返回：

一个2-D Tensor，数据类型为int64，包含所有输入中的非零数值的位置。

异常：

• TypeError - 如果 input 不是Tensor。

• ValueError - 如果 input 的维度等于0。

支持平台：

Ascend GPU CPU

样例：

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