计算对数的keras层?

LWi*_*son 1 keras keras-layer

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

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

Dan*_*ler 5

您可以在Lambda层内使用任何后端函数:

from keras.layers import Lambda
import keras.backend as K
Run Code Online (Sandbox Code Playgroud)

定义任何采用输入张量的函数:

def logFunc(x):
   return K.log(x)
Run Code Online (Sandbox Code Playgroud)

并用它创建一个 lambda 层:

from keras.layers import Lambda
import keras.backend as K
Run Code Online (Sandbox Code Playgroud)

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