VariationalAutoEncoder
======================

.. currentmodule:: deeplay.applications.autoencoders.vae

.. autoclass:: VariationalAutoEncoder
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~VariationalAutoEncoder.compute_loss
      ~VariationalAutoEncoder.decode
      ~VariationalAutoEncoder.encode
      ~VariationalAutoEncoder.forward
      ~VariationalAutoEncoder.reparameterize
      ~VariationalAutoEncoder.training_step

   .. rubric:: Methods Documentation

   .. automethod:: compute_loss
   .. automethod:: decode
   .. automethod:: encode
   .. automethod:: forward
   .. automethod:: reparameterize
   .. automethod:: training_step