WassersteinAutoEncoder
======================

.. currentmodule:: deeplay.applications.autoencoders.wae

.. autoclass:: WassersteinAutoEncoder
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~WassersteinAutoEncoder.compute_IMQ
      ~WassersteinAutoEncoder.compute_kernel
      ~WassersteinAutoEncoder.compute_loss
      ~WassersteinAutoEncoder.compute_mmd
      ~WassersteinAutoEncoder.decode
      ~WassersteinAutoEncoder.encode
      ~WassersteinAutoEncoder.forward
      ~WassersteinAutoEncoder.training_step

   .. rubric:: Methods Documentation

   .. automethod:: compute_IMQ
   .. automethod:: compute_kernel
   .. automethod:: compute_loss
   .. automethod:: compute_mmd
   .. automethod:: decode
   .. automethod:: encode
   .. automethod:: forward
   .. automethod:: training_step