Imported from https://github.com/ChengBinJin/WGAN-GP-tensorflow