CANN ops-nn Sigmoid梯度算子

SigmoidGrad

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn

产品支持情况

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

功能说明

• 接口功能：完成sigmoid的反向传播，根据sigmoid反向传播梯度与正向输出计算sigmoid的梯度输入。

• 计算公式：

  $$ out = {\frac{1} {1+{e}^{-input}}} $$

  $$ grad_input = grad_output * \sigma(x) * (1 - \sigma(x)) $$

其中$\sigma(x)$为sigmoid函数的正向输出，$\sigma(x)*(1-\sigma(x))$为sigmoid函数的导数。

参数说明

参数名	输入/输出	描述	使用说明	数据类型	数据格式	维度(shape)	非连续Tensor
gradOutput	输入	损失函数对sigmoid输出的梯度，公式中的grad\_output。	支持空Tensor。 shape需要与output，grad_input一致。	FLOAT、FLOAT16、BFLOAT16、DOUBLE、COMPLEX64、COMPLEX128	ND	1-8	√
output	输入	前向sigmoid的输出，公式中的out。	支持空Tensor。 shape需要与grad_output，grad_input一致。	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算子。

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn