Variational Autoencoder: Practice

It is finally time to put everything we learned together and build the infamous variational autoencoder.

Here is an overview of what we are going to implement:

For the decoder, we will use a simple network with 2 linear layers. This network will parameterize the variational posterior $q_{\phi}(z|x)$ ...