mindspore.ops.where

mindspore.ops.where(condition, input, other)

返回一个tensor，其中元素根据 condition 从 input 或 other 中选择。

支持广播。

$$\begin{array}{r}outpu{t}_i=\{\begin{array}{ll}inpu{t}_i,& ifconditio{n}_i\\ othe{r}_i,& otherwise\end{array}\end{array}$$

参数：

• condition (Tensor[bool]) - 如果为 True ，选取 input 中的元素，否则选取 other 中的元素。

• input (Union[Tensor, Scalar]) - 在 condition 为 True 的索引处选择的值。

• other (Union[Tensor, Scalar]) - 当 condition 为 False 的索引处选择的值。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.tensor([[0, 1],
...                           [2, 3]])
>>> other = mindspore.tensor([[1, 1],
...                           [1, 1]])
>>> condition = input < 3
>>> mindspore.ops.where(condition, input, other)
Tensor(shape=[2, 2], dtype=Int64, value=
[[0, 1],
 [2, 1]])