CANN/ops-nn: Hardtanh梯度算子

发布时间:2026/7/13 3:08:51

CANN/ops-nn: Hardtanh梯度算子 HardtanhGrad【免费下载链接】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 训练系列产品√功能说明算子功能激活函数Hardtanh的反向。计算公式$$ grad_self_{i} \begin{cases} 0,\ \ \ \ \ \ \ if \ \ self_{i}max \ 0,\ \ \ \ \ \ \ if\ \ self_{i}min \ 1,\ \ \ \ \ \ \ \ \ \ \ \ otherwise \ \end{cases} $$$$ res_{i} grad_output_{i} \times grad_self_{i} $$参数说明参数名输入/输出/属性描述数据类型数据格式result输入反向传播过程中上一步输出的梯度公式中的grad_output。BFLOAT16、FLOAT16、FLOATNDgrad输入正向的输入数据公式中的self。BFLOAT16、FLOAT16、FLOATNDmin_val属性线性范围的下限公式中的min。FLOAT-max_val属性线性范围的上限公式中的max。FLOAT-out输出计算得到梯度公式中的res。BFLOAT16、FLOAT16、FLOATND约束说明result、grad和out的shape和数据类型需要一致。调用说明调用方式调用样例说明aclnn调用test_aclnn_hardtanh_grad通过aclnnHardtanhBackWard方式调用HardtanhGrad算子。图模式-通过算子IR构图方式调用HardtanhGrad算子。【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

相关新闻