InvGrad

产品支持情况

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

功能说明

  • 算子功能:计算 Inv(取倒数)算子的反向梯度,对应前向 y = 1 / x 的梯度计算。

  • 计算公式:

    dxi=−dyi⋅yi2dx_i = -dy_i \cdot y_i^2

参数说明

参数名 输入/输出 描述 数据类型 数据格式
y 输入
  • 公式中的输入 y(前向 Inv 的输出)。
  • 支持空 Tensor。
  • 支持非连续 Tensor。
  • shape 维度:0-8。
FLOAT、FLOAT16、BFLOAT16 ND
dy 输入
  • 公式中的输入 dy(下游传回的梯度)。
  • shape 和 dtype 必须与 y 完全一致。
FLOAT、FLOAT16、BFLOAT16 ND
dx 输出
  • 公式中的输出 dx(对前向输入 x 的梯度)。
  • shape 和 dtype 必须与 y/dy 完全一致。
FLOAT、FLOAT16、BFLOAT16 ND