文件最后提交记录最后更新时间
21 小时前
18 天前
18 天前
18 天前
18 天前
18 天前
18 天前
2 个月前
1 天前
README

HardSigmoidGrad

产品支持情况

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

功能说明

  • 算子功能:HardSigmoid的反向算子,计算反向传播的梯度grad_input。

  • 计算公式:

    HardsigmoidBackward(self,grad_output)={grad_output∗α,if(0<α∗self+β<1)0,otherwiseHardsigmoidBackward(self, grad\_output)= \begin{cases} &grad\_output \ast \alpha, &if(0 < \alpha \ast self + \beta < 1) \\ &0, &otherwise \end{cases}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
grads 输入 反向传播过程中上一步输出的梯度,作为本反向算子的输入,公式中的grad_output。 FLOAT、FLOAT16、BFLOAT16 ND
input_x 输入 表示HardSigmoid前向算子的输入Tensor,公式中的self。 FLOAT、FLOAT16、BFLOAT16 ND
y 输出 反向传播梯度输出,公式中的HardsigmoidBackward(self, grad_output)。 FLOAT、FLOAT16、BFLOAT16 ND
  • 公式中的α\alpha固定为1/61/6β\beta固定为0.50.5

约束说明

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_hard_sigmoid_grad.cpp 通过aclnnHardsigmoidBackward接口方式调用HardSigmoidGrad算子。