CANN/ops-nn Tanh梯度算子

TanhGrad【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn产品支持情况产品是否支持Atlas A2 训练系列产品/Atlas 800I A2 推理产品√功能说明算子功能计算 Tanh 的梯度。计算公式$$ dx dy * (1 - y * y) $$其中 $y$ 为 Tanh 前向输出$dy$ 为上游梯度$dx$ 为输入梯度。参数说明参数名输入/输出/属性描述数据类型数据格式y输入Tanh 前向输出公式中的 y。FLOAT、FLOAT16、BFLOAT16NDdy输入上游梯度公式中的 dy。FLOAT、FLOAT16、BFLOAT16NDdx输出输入梯度公式中的 dx。FLOAT、FLOAT16、BFLOAT16ND约束说明输入 y、dy 和输出 dx 的数据类型需保持一致。输入 y 和 dy 的 shape 需相同。贡献说明贡献者贡献方贡献算子贡献时间贡献内容Zhou Jianhua LePenseurAISS Group, Harbin Institute of Technology (HIT)TanhGrad2026/05/12TanhGrad 算子架构迁移至标准 AscendC 框架【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考