from keras.layers.core import Lambda
from keras.layers import Input
def function(x):
return x*x
input_img=Input(input_shape=[277,277,3])
m=Lambda(function,out_shape=lambda input_shape: input_shape)(input_img)
from keras.layers.core import Lambda
from keras.layers import Input
def function(x):
return x*x
input_img=Input(input_shape=[277,277,3])
m=Lambda(function,out_shape=lambda input_shape: input_shape)(input_img)