mindspore.ops.where
====================

.. py:function:: mindspore.ops.where(condition, input, other)

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

    支持广播。

    .. math::

        output_i = \begin{cases} input_i,\quad &if\ condition_i \\ other_i,\quad &otherwise \end{cases}

    .. warning::
        从2.9.0（不含）之后版本开始，该接口将支持单参数重载 ``mindspore.ops.where(condition)`` ，
        仅传入 `condition` 时将返回非零元素的坐标，等价于 ``mindspore.ops.argwhere(condition)`` 。
        该变更与当前版本兼容，三参数调用时行为不变。

    参数：
        - **condition** (Tensor[bool]) - 如果为 ``True`` ，选取 `input` 中的元素，否则选取 `other` 中的元素。
        - **input** (Union[Tensor, Scalar]) - 在 `condition` 为 ``True`` 的索引处选择的值。
        - **other** (Union[Tensor, Scalar]) - 当 `condition` 为 ``False`` 的索引处选择的值。

    返回：
        Tensor