Imported from https://github.com/Natsu6767/InfoGAN-PyTorch