20 lines
873 B
ReStructuredText
20 lines
873 B
ReStructuredText
mindspore.numpy.where
|
||
=================================
|
||
|
||
.. py:function:: mindspore.numpy.where(condition, x=None, y=None)
|
||
|
||
根据 ``condition`` 从 ``x`` 或 ``y`` 中选择元素。
|
||
|
||
.. note::
|
||
由于不支持 ``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。 |