Imported from https://github.com/y0ast/VAE-TensorFlow