Keras自定义层函数

使用keras时候经常会自定义一些函数，而Keras由于封装的过于完整，所以需要利用Lambda函数自定义写层函数

导入相关库

from keras import backend as K
from keras.layers.core import Lambda

自定义函数

def function(x, a):
    return (a+1)*K.abs(x) / (a + K.abs(x))

其中a是参数

调用

y= Lambda(function, output_shape=list(x.shape), arguments={'a':100})(x)