EluGrad【免费下载链接】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 训练系列产品√功能说明算子功能对应Elu操作的反向传播梯度。计算公式$$ y (min(activations,0) 1)* grads $$参数说明参数名输入/输出/属性描述数据类型数据格式grads输入对应Elu操作的反向传播梯度FLOAT、FLOAT16、BFLOAT16NDactivations输入与grads具有相同的类型、格式和形状。对应Elu操作的输出值。FLOAT、FLOAT16、BFLOAT16NDy输出公式中的输出张量同gradsND调用说明调用方式样例代码说明图模式test_geir_elu_backward.cpp通过算子IR构图方式调用EluGrad算子。【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考