Imported from https://github.com/kuc2477/pytorch-vae