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、BFLOAT16 | ND |
| dy | 输入 | 上游梯度，公式中的 dy。 | FLOAT、FLOAT16、BFLOAT16 | ND |
| dx | 输出 | 输入梯度，公式中的 dx。 | FLOAT、FLOAT16、BFLOAT16 | ND |
约束说明
- 输入 y、dy 和输出 dx 的数据类型需保持一致。
- 输入 y 和 dy 的 shape 需相同。
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| Zhou Jianhua <@LePenseur> | AISS Group, Harbin Institute of Technology (HIT) | TanhGrad | 2026/05/12 | TanhGrad 算子架构迁移至标准 AscendC 框架 |
【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考