credit.postblock.base
=====================

.. py:module:: credit.postblock.base


Classes
-------

.. autoapisummary::

   credit.postblock.base.BasePostblock


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

.. py:class:: BasePostblock(*args: Any, **kwargs: Any)

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


   Base class for all postblocks.

   Forward signature for all postblocks::

       forward(batch: dict) -> dict

   where ``batch`` is the reconstructed dict returned by ``Reconstruct``::

       {
           "prediction": {var_key: tensor, ...},
           "metadata":   {...},
       }


   .. py:method:: forward(batch: dict) -> dict


   .. py:method:: from_config(**kwargs)
      :classmethod: