timesead.models.reconstruction.autoformer
=========================================

.. py:module:: timesead.models.reconstruction.autoformer


Classes
-------

.. autoapisummary::

   timesead.models.reconstruction.autoformer.Autoformer


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

.. py:class:: Autoformer(window_size: int, input_dim: int, moving_avg: int = 25, model_dim: int = 128, dropout: float = 0.1, attention_factor: int = 1, num_heads: int = 8, fcn_dim: int = 128, activation: str = 'gelu', encoder_layers: int = 3)

   Bases: :py:obj:`timesead.models.BaseModel`


   Autoformer is the first method to achieve the series-wise connection,
   with inherent O(LlogL) complexity
   Paper link: https://openreview.net/pdf?id=I55UqU-M11y

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


   .. py:attribute:: seq_len


   .. py:attribute:: decomp


   .. py:attribute:: enc_embedding


   .. py:attribute:: encoder


   .. py:attribute:: projection


   .. py:method:: forward(inputs: Tuple[torch.Tensor, Ellipsis]) -> Tuple[torch.Tensor, Ellipsis]