Autoencoder And Variational Autoencoder Pytorch