autoencoder for dimensionality reduction pytorch