"""

tf.keras.layers.Lambda(

    function, output_shape=None, mask=None, arguments=None, **kwargs

)

"""

def func(x):

    return np.array(x) + 3

​

x1 = keras.layers.Lambda(func)([[1,2,3,4,5]])

print(x1)

>>> [[4 5 6 7 8]]

​

# Custom Guassian Noise Layer

x2 = keras.layers.Lambda(lambda x: # input => x

                         x + tf.random.normal(tf.shape(x), stddev=2) # <= output

                        )