MaskedScale

产品支持情况

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

功能说明

• 算子功能：完成elementwise掩码缩放计算。

• 计算公式：

  $$y=x\times mask\times scale$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入张量，公式中的x。	FLOAT16、BFLOAT16、FLOAT	ND
mask	输入	掩码张量，公式中的mask，shape与x一致。	INT8、UINT8、FLOAT16、FLOAT	ND
scale	属性	缩放系数，公式中的scale。	FLOAT	-
y	输出	输出张量，公式中的y，数据类型和shape与x一致。	FLOAT16、BFLOAT16、FLOAT	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_masked_scale	通过aclnnMaskedScale接口方式调用MaskedScale算子。