zl程序教程

您现在的位置是:首页 >  工具

当前栏目

深度学习7 hsigmoid硬件友好的激活函数实现

学习硬件 实现 函数 深度 激活 友好
2023-09-14 09:15:04 时间

hsigmoid激活函数是对sigmoid的近似,尽可能对sigmoid进行拟合。值域为【0,1】,斜率为1/6

公式为:

,相比于sigmoid的指数运算可以节省大量的计算时间。

基于relu函数实现,小于0置0,大于1置1

from keras.utils.generic_utils import get_custom_objects
tmp=0
def hsigmoid(x):
    x=x/6+0.5
    x=K.relu(x,max_value=1)*x/x
    return x
get_custom_objects().update({'hsigmoid': layers.Activation(hsigmoid)})

训练试用

使用hsigmoid训练
使用hsigmoid训练网络