2022-02-19 15:19:19 +08:00
|
|
|
|
mindspore.ops.MaskedFill
|
|
|
|
|
|
=========================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.ops.MaskedFill()
|
|
|
|
|
|
|
|
|
|
|
|
掩码为True位置的值将被指定的值填充。
|
|
|
|
|
|
|
2022-02-26 11:08:21 +08:00
|
|
|
|
`input` 和 `mask` 的shape需相同或可广播。
|
2022-02-19 15:19:19 +08:00
|
|
|
|
|
|
|
|
|
|
**输入：**
|
|
|
|
|
|
|
|
|
|
|
|
- **input** (Tensor) - 输入Tensor，其数据类型为float16、float32、int8、或int32。
|
|
|
|
|
|
- **mask** (Tensor[bool]) - 输入的掩码，其数据类型为bool。
|
|
|
|
|
|
- **value** (Union[float, Tensor]) - 用来填充的值，只支持0维Tensor或浮点数。
|
|
|
|
|
|
|
|
|
|
|
|
**输出：**
|
|
|
|
|
|
|
|
|
|
|
|
Tensor，输出与输入的数据类型和shape相同。
|
|
|
|
|
|
|
|
|
|
|
|
**异常：**
|
|
|
|
|
|
|
|
|
|
|
|
- **TypeError** - `input` 或 `mask` 不是Tensor。
|
|
|
|
|
|
- **TypeError** - `value` 既不是浮点数也不是Tensor。
|
|
|
|
|
|
- **TypeError** - `input` 或 `value` 的数据类型不是float16、float32、int8、或int32。
|
|
|
|
|
|
- **TypeError** - `value` 的数据类型与 `input` 不同。
|
|
|
|
|
|
- **TypeError** - `mask` 的数据类型不是bool。
|
|
|
|
|
|
- **ValueError** - `input` 和 `mask` 的shape不可广播。
|
