Imported from https://github.com/alectryonexamples/gan_tf2