Imported from https://github.com/ChengBinJin/VAE-Tensorflow