Keras: Lambda layer function with multiple parameters

python neural-network keras keras-layer
Prabaha picture Prabaha · Jul 5, 2017 · Viewed 11.7k times · Source

I am trying to write a Lambda layer in Keras which calls a function connection, that runs a loop for i in range(0,k) where k is fed in as an input to the function, connection(x,k). Now, when I try to call the function in the Functional API, I tried using:

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

Also,

y = Lambda(connection)(x,k)

But neither of those approaches worked. How can I feed in the value of k without assigning it as a global parameter?

Answer

Andrey Nikishaev picture Andrey Nikishaev · Jul 9, 2018

Just use 

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

and then var[0], var[1] in connection method

Related questions

Negative dimension size caused by subtracting 3 from 1 for 'conv2d_2/convolution'

I got this error message when declaring the input layer in Keras. ValueError: Negative dimension size caused by subtracting 3 from 1 for 'conv2d_2/convolution' (op: 'Conv2D') with input shapes: [?,1,28,28], [3,3,28,32]. My code is like this model.add(Convolution2D(32, 3, 3, activation=…

Read More
How to implement Merge from Keras.layers

I have been trying to merge the following sequential models but haven't been able to. Could somebody please point out my mistake, thank you. The code compiles while using"merge" but give the following error "TypeError: 'module' object is not …

Read More
element-wise multiplication with broadcasting in keras custom layer

I am creating a custom layer with weights that need to be multiplied by element-wise before activation. I can get it to work when the output and input is the same shape. The problem occurs when I have a first …

Read More