mindspore.numpy.where

mindspore.numpy.where(condition, x=None, y=None)

根据 condition 从 x 或 y 中选择元素。

说明

由于不支持 nonzero ， x 和 y 必须都是Tensor输入。

参数：

• condition (Tensor) - 当为 True 时，选择 x 中的值，否则选择 y 中的值。

• x (Tensor，可选) - 选择值的来源。默认值： None 。

• y (Tensor，可选) - 选择值的来源。 x ， y 和 condition 需要能够广播到相同的shape。默认值： None 。

返回：

Tensor或标量，其中 condition 为 True 的位置取自 x ，其他位置取自 y 。

异常：

• ValueError - 如果操作数不能广播到相同的shape。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> condition = np.full((1, 1, 2), [False, True])
>>> x = np.full((1, 3, 2), 5)
>>> y = np.full((2, 1, 1), 7)
>>> output = np.where(condition, x, y)
>>> print(output)
[[[7 5]
[7 5]
[7 5]]
[[7 5]
[7 5]
[7 5]]]