SigmoidGrad
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
-
接口功能:完成sigmoid的反向传播,根据sigmoid反向传播梯度与正向输出计算sigmoid的梯度输入。
-
计算公式:
out=11+e−inputout = {\frac{1} {1+{e}^{-input}}}
grad_input=grad_output∗σ(x)∗(1−σ(x))grad\_input = grad\_output * \sigma(x) * (1 - \sigma(x))
其中σ(x)\sigma(x)为sigmoid函数的正向输出,σ(x)∗(1−σ(x))\sigma(x)*(1-\sigma(x))为sigmoid函数的导数。
参数说明
| 参数名 | 输入/输出 | 描述 | 使用说明 | 数据类型 | 数据格式 | 维度(shape) | 非连续Tensor |
|---|---|---|---|---|---|---|---|
| gradOutput | 输入 | 损失函数对sigmoid输出的梯度,公式中的grad\_output。 |
|
FLOAT、FLOAT16、BFLOAT16、DOUBLE、COMPLEX64、COMPLEX128 | ND | 1-8 | √ |
| output | 输入 | 前向sigmoid的输出,公式中的out。 |
|
FLOAT、FLOAT16、BFLOAT16、DOUBLE、COMPLEX64、COMPLEX128 | ND | 1-8 | √ |
| gradInput | 输出 | 为self的梯度值,公式中的grad\_input。 | 数据类型、shape需要与grad_output,output一致。 | FLOAT、FLOAT16、BFLOAT16、DOUBLE、COMPLEX64、COMPLEX128 | ND | 1-8 | √ |
- Atlas 推理系列产品、Atlas 训练系列产品:数据类型支持FLOAT、FLOAT16、DOUBLE、COMPLEX64、COMPLEX128。
约束说明
无
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn接口 | test_aclnn_sigmoid_grad.cpp | 通过aclnnSigmoidBackward接口方式调用SigmoidGrad算子。 |
| 图模式 | - | 通过算子IR构图方式调用SigmoidGrad算子。 |