2017-10-05 58 views
1

我想建立一个Keras层,其中每个节点只需计算上一层中对应节点的对数。我从Keras文档中看到它的后端模块中有一个“日志”功能。但不知何故,我不知道如何使用它。计算对数的keras层?

在此先感谢您提供的任何提示!

回答

0

你可以使用任何后端功能的Lambda层内:

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

定义只是任何函数取输入张量:

def logFunc(x): 
    return K.log(x) 

,并创建一个lambda层:

#add to the model the way you're used to: 
model.add(Lambda(logFunc,output_shape=(necessaryWithTheano))) 
+0

谢谢!我一直在使用Keras Sequential API构建我的网络,而不是功能性API。我将如何处理您使用Sequential API的建议? – LWixson

+0

'model.add(Lambda(.....))' –