Autoencoder Pytorch Example Transformer