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