tf.keras.layers.BatchNormalization

Layer that normalizes its inputs.

Inherits From: Layer, Module

Used in the notebooks

Used in the guide Used in the tutorials

Batch normalization applies a transformation that maintains the mean output close to 0 and the output standard deviation close to 1.

Importantly, batch normalization works differently during training and during inference.

During training (i.e. when using fit() or when calling the layer/model with the argument training=True), the layer normalizes its output using the mean and standard deviation of the current batch of inputs. That is to say, for each channel being normalized, the layer returns gamma * (batch - mean