Imported from https://github.com/GrayXu/SpectralNormalization-TF-Keras