looking for an equivalent of Tensorflow normalization layer in Pytorch

Question

I was using \'tf.keras.layers.experimental.preprocessing.Normalization\'. This layer is cool since you can save weights in this layer to normalize any input data to this layer.