CANN/ops-nn ReLU梯度V3算子

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_outputz梯度计算结果grad_input当 x 0 时等于 y否则为 0参数说明参数名输入/输出/属性描述数据类型数据格式x输入前向ReLU的输入。fp32/fp16/bf16/int32/uint8/int8NDy输入反向传播的梯度输入grad_output。fp32/fp16/bf16/int32/uint8/int8NDz输出梯度计算结果输出grad_inputshape与x一致。fp32/fp16/bf16/int32/uint8/int8ND约束说明输入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)ReluGradV32026/5/13ReluGradV3算子适配开源仓多核tiling、CompareScalar/Select API、aclnn测试通过【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考