timesead.models.common.ae
=========================

.. py:module:: timesead.models.common.ae


Classes
-------

.. autoapisummary::

   timesead.models.common.ae.AE


Module Contents
---------------

.. py:class:: AE(encoder: torch.nn.Module, decoder: torch.nn.Module, return_latent: bool = False)

   Bases: :py:obj:`torch.nn.Module`


   Simple AE Implementation

   Initialize internal Module state, shared by both nn.Module and ScriptModule.


   .. py:attribute:: encoder


   .. py:attribute:: decoder


   .. py:attribute:: return_latent
      :value: False



   .. py:method:: forward(x: torch.Tensor) -> Union[torch.Tensor, Tuple[torch.Tensor, torch.Tensor]]