Keras example - building a custom normalization layer