Variational Auto Encoder Latent dim_3