Autoencoder Example Pytorch Code