文件最后提交记录最后更新时间
4 个月前
2 个月前
4 个月前
4 个月前
4 个月前
3 个月前
README

RsqrtGrad

产品支持情况

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

功能说明

  • 算子功能:完成梯度平方根计算。
  • 计算公式:

z=y∗y∗y∗dy∗(−0.5)z = y * y * y * dy * (-0.5)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
y 输入 待进行rsqrt_grad计算的入参,公式中的y。 DT_FLOAT,DT_FLOAT_16,DT_BF16 ND
dy 输入 待进行rsqrt_grad计算的入参,公式中的dy。 DT_FLOAT,DT_FLOAT_16,DT_BF16 ND
z 输出 待进行rsqrt_grad计算的出参,公式中的z。 DT_FLOAT,DT_FLOAT_16,DT_BF16 ND

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_rsqrt_grad 使用自动生成的aclnn接口调用RsqrtGrad算子。