Implement normalization in a layer

normalize.py

class Normalize(layers.Layer):
    """ Custom Layer for Preprocessing Input """
    def __init__(self):
        """ Constructor """
        super(Normalize, self).__init__()
    
    def build(self, input_shape):
        """ Handler for Input Shape """
        self.kernel = None
    
    @tf.function
    def call(self, inputs):
        """ Handler for layer object is callable """
        inputs = inputs / 255.0
        return inputs