popgym.baselines.models.aggregations
====================================

.. py:module:: popgym.baselines.models.aggregations


Classes
-------

.. autoapisummary::

   popgym.baselines.models.aggregations.Aggregation
   popgym.baselines.models.aggregations.SumAggregation
   popgym.baselines.models.aggregations.MaxAggregation


Functions
---------

.. autoapisummary::

   popgym.baselines.models.aggregations.get_aggregator


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

.. py:function:: get_aggregator(name: str) -> torch.nn.Module

.. py:class:: Aggregation(*args, **kwargs)

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


   Aggregates (x_k ... x_t , s_k) into s_t


   .. py:method:: forward(x: torch.Tensor, memory: torch.Tensor) -> torch.Tensor
      :abstractmethod:



.. py:class:: SumAggregation(*args, **kwargs)

   Bases: :py:obj:`Aggregation`


   Aggregates (x_k ... x_t , s_k) into s_t


   .. py:method:: forward(x: torch.Tensor, memory: torch.Tensor) -> torch.Tensor


.. py:class:: MaxAggregation(*args, **kwargs)

   Bases: :py:obj:`Aggregation`


   Aggregates (x_k ... x_t , s_k) into s_t


   .. py:method:: forward(x: torch.Tensor, memory: torch.Tensor) -> torch.Tensor