Imported from https://github.com/jessemzhang/dl_spectral_normalization