文件最后提交记录最后更新时间
30 天前
18 天前
30 天前
30 天前
30 天前
30 天前
30 天前
16 天前
README

SoftplusGrad

产品支持情况

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

功能说明

接口功能:Softplus激活函数的反向传播,对齐TensorFlow raw_ops.SoftplusGrad

计算公式:

backprops=gradients×sigmoid(features)=gradients1+exp(−features)backprops = gradients \times sigmoid(features) = \frac{gradients}{1 + exp(-features)}

支持 NumPy 风格广播:gradients 和 features 可以有不同的 shape,输出 shape 为广播后的结果。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
gradients 输入 表示梯度。 FLOAT、FLOAT16、BFLOAT16 ND
features 输入 表示输入张量。 FLOAT、FLOAT16、BFLOAT16 ND
backprops 输出 表示反向传播结果。 FLOAT、FLOAT16、BFLOAT16 ND

约束说明

  • gradients、features、backprops的数据类型必须相同。
  • gradients和features支持NumPy风格广播。
  • 支持0-8维输入。

调用说明

调用方式 样例代码 说明
图模式 test_geir_softplus_grad.cpp 通过GE IR构建图方式调用SoftplusGrad算子。