ReluGradV3
【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A2 训练系列产品/Atlas 800I A2 推理产品 | √ |
功能说明
算子功能：求ReLU函数的梯度。
计算公式：
$$ z = (x > 0) \ ? \ y : 0 $$
其中：
x：前向ReLU的输入y：来自上游的反向传播梯度（grad_output）z：梯度计算结果（grad_input），当 x > 0 时等于 y，否则为 0
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 前向ReLU的输入。 | fp32/fp16/bf16/int32/uint8/int8 | ND |
| y | 输入 | 反向传播的梯度输入（grad_output）。 | fp32/fp16/bf16/int32/uint8/int8 | ND |
| z | 输出 | 梯度计算结果输出（grad_input），shape与x一致。 | fp32/fp16/bf16/int32/uint8/int8 | ND |
约束说明
- 输入
x、y及输出z的数据类型必须一致，支持 fp32、fp16、bf16。 - 输入
y支持与xshape 完全相同，或作为单元素标量广播到x的 shape。 - 输出
z的 shape 与输入x一致。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_relu_grad_v3.cpp | 通过aclnnReluGradV3接口方式调用ReluGradV3算子。 |
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| Shi Xiangyang (shi-xiangyang225) | AISS Group, Harbin Institute of Technology (HIT) | ReluGradV3 | 2026/5/13 | ReluGradV3算子适配开源仓（多核tiling、CompareScalar/Select API、aclnn测试通过） |
【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
