Imported from https://github.com/manicman1999/StyleGAN2-Tensorflow-2.0