Why do we pass input X to a layer like BatchNormalization like this : X = BatchNormalization(axis = 3)(X)

Maybe take a look at this thread and let us know what you discover ?