Autoencoder Example Pytorch