Also in the VAE diagram, the latent space appears to have the same dimensions as input and output.