SqrtGrad

产品支持情况

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

功能说明

• 算子功能：完成梯度平方根计算。

• 计算公式：

$$z=dy\ast 0.5/y$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
y	输入	待进行sqrt_grad计算的入参，公式中的y。	BFLOAT16、FLOAT16、FLOAT	ND
dy	输入	待进行sqrt_grad计算的入参，公式中的dy。	BFLOAT16、FLOAT16、FLOAT	ND
z	输出	待进行sqrt_grad计算的出参，公式中的z。	BFLOAT16、FLOAT16、FLOAT	ND

约束说明

无

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_sqrt_grad	通过算子IR构图方式调用SqrtGrad算子。