forked from mindspore-Ecosystem/mindspore
13 lines
651 B
ReStructuredText
13 lines
651 B
ReStructuredText
mindspore.ops.MaskedFill
|
||
=========================
|
||
|
||
.. py:class:: mindspore.ops.MaskedFill
|
||
|
||
将掩码位置为True的位置填充指定的值。
|
||
|
||
.. note::
|
||
如果 `value` 是Python类型的浮点数，则默认会转为float32类型。这种情况下，当 `input_x` 为float16类型时，在CPU和Ascend平台上，`input_x` 会转为float32类型参与计算，
|
||
并将结果类型转换到float16类型，可能会造成一定性能损耗；而在GPU平台上，则会引起TypeError。因此建议 `value` 采用与 `input_x` 具有相同数据类型的Tensor。
|
||
|
||
更多参考详见 :func:`mindspore.ops.masked_fill`。
|