LeakyRelu【免费下载链接】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 训练系列产品√功能说明接口功能激活函数用于解决Relu函数在输入小于0时输出为0的问题避免神经元无法更新参数。计算公式$$ out max(0,self) negativeSlope * min(0,self) $$参数说明参数名输入/输出/属性描述数据类型数据格式self输入待进行LeakyRelu激活函数的入参公式中的self。FLOAT、FLOAT16、BFLOAT16NDnegativeSlope输入表示self 0时的斜率公式中的negativeSlope。FLOAT-out输出待进行LeakyRelu激活函数的出参。FLOAT、FLOAT16、BFLOAT16NDAtlas 训练系列产品 FLOAT、FLOAT16。约束说明negativeSlope使用整型类型作为属性输入而输入self是FLOAT类型那么如果negativeSlope大于2^24或小于-2^24可能存在精度损失。同理如果输入self是FLOAT16类型那么negativeSlope大于2^11或小于-2^11可能存在精度损失。调用说明调用方式样例代码说明aclnn接口test_aclnn_leaky_relu.cpp通过aclnnLeakyReluaclnnInplaceLeakyRelu接口方式调用LeakyRelu算子。aclnn接口test_aclnn_inplace_leaky_relu.cpp通过aclnnLeakyReluaclnnInplaceLeakyRelu接口方式调用LeakyRelu算子。【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考