As we are using TF dataset objects to stream data, we need to reconsider how to implement normalisation of input to the VAE