==================
vae.py
==================
.. automodule:: vae
   :members:
   :undoc-members:   
   :special-members: