Keras:具有多个参数的Lambda层功能

Question

我正在尝试Lambda在Keras中编写一个调用函数的图层,该图层connection运行一个循环for i in range(0,k),其中k作为函数的输入被输入connection(x,k).现在,当我尝试在Functional API中调用该函数时,我尝试使用:

k = 5
y = Lambda(connection)(x)

也,

y = Lambda(connection)(x,k)

但这些方法都没有奏效.如何在k不将其指定为全局参数的情况下输入值？

Answer 1

只是用

y = Lambda(connection)((x,k)) 

然后在连接方法中使用var [0],var [1]

Answer 2

在此GitHub Pull Request中找到了问题的解决方案.运用

y = Lambda(connection, arguments={'k':k})(x)

成功了!