计算对数的keras层？

Question

我想设置一个 Keras 层，其中每个节点只计算前一层中相应节点的对数。我从 Keras 文档中看到它的后端模块中有一个“日志”功能。但不知何故，我不明白如何使用它。

提前感谢您提供的任何提示!

Answer 1

您可以在 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)))

如果函数已经定义，只接受一个参数并返回一个张量，则您无需创建自己的函数，例如 Lambda(K.log)。