FilesLast commitLast update
2 months ago
2 months ago
20 days ago
2 months ago
2 months ago
20 days ago
1 month ago
README

ReciprocalGrad

产品支持情况

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

功能说明

  • 算子功能:计算 Reciprocal 算子的梯度。其中 y = Reciprocal(x) = 1/x,dy 为上游梯度,z 为对输入 x 的梯度。
  • 计算公式:

z=−y×y×dyz = -y \times y \times dy

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
y 输入 公式中的y,前向 Reciprocal 算子的输出。 FLOAT、FLOAT16、BFLOAT16 ND
dy 输入 公式中的dy,上游梯度。 FLOAT、FLOAT16、BFLOAT16 ND
z 输出 公式中的z,对输入x的梯度。 FLOAT、FLOAT16、BFLOAT16 ND
  • Atlas训练系列产品、Atlas推理系列产品:不支持BFLOAT16。

约束说明

  • y、dy、z 三者的数据类型必须一致。
  • y、dy、z 三者的 shape 必须一致。
  • 数据格式仅支持 ND。
  • shape 维度不大于 8 维。

调用说明

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