2022-02-19 15:19:19 +08:00
|
|
|
|
mindspore.ops.MaskedFill
|
|
|
|
|
|
=========================
|
|
|
|
|
|
|
2022-11-19 10:12:44 +08:00
|
|
|
|
.. py:class:: mindspore.ops.MaskedFill
|
2022-02-19 15:19:19 +08:00
|
|
|
|
|
2022-03-29 17:06:07 +08:00
|
|
|
|
将掩码位置为True的位置填充指定的值。
|
2022-02-19 15:19:19 +08:00
|
|
|
|
|
2022-11-16 23:03:34 +08:00
|
|
|
|
.. note::
|
|
|
|
|
|
如果 `value` 是Python类型的浮点数,则默认会转为float32类型。这种情况下,当 `input_x` 为float16类型时,在CPU和Ascend平台上,`input_x` 会转为float32类型参与计算,
|
|
|
|
|
|
并将结果类型转换到float16类型,可能会造成一定性能损耗;而在GPU平台上,则会引起TypeError。因此建议 `value` 采用与 `input_x` 具有相同数据类型的Tensor。
|
|
|
|
|
|
|
2022-06-04 15:32:43 +08:00
|
|
|
|
更多参考详见 :func:`mindspore.ops.masked_fill`。
|