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