MaskedFill

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：将输入Tensorx中mask位置为True的元素填充指定的值。mask必须与x的shape相同或可广播。

• 计算公式：

  $$y_i=\{\begin{array}{l}{x}_i,mas{k}_i=False\\ valu{e}_i,mas{k}_i=True\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	公式中的输入x。	FLOAT32、FLOAT16、BFLOAT16、INT8、INT32、BOOL	ND
mask	输入	公式中的输入mask。	BOOL	ND
value	输入	公式中的输入value。	FLOAT32、FLOAT16、BFLOAT16、INT8、INT32、BOOL	ND
y	输出	公式中的输出y。	FLOAT32、FLOAT16、BFLOAT16、INT8、INT32、BOOL	ND

• Atlas 训练系列产品、Atlas 推理系列产品、Atlas 200I/500 A2 推理产品：不支持BFLOAT16。

约束说明

• value的dtype与x不一致时，需保证进行类型转换时不会溢出。
• x和mask需要满足广播关系。
• value的数据类型能够转换为x的数据类型。
• y的shape可以由x、mask、value共同广播。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_inplace_masked_fill_scalar	通过aclnnInplaceMaskedFillScalar接口方式调用MaskedFill算子。