:py:mod:`rofunc.learning.RofuncRL.models.base_models`
=====================================================

.. py:module:: rofunc.learning.RofuncRL.models.base_models

.. autodoc2-docstring:: rofunc.learning.RofuncRL.models.base_models
   :allowtitles:

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

Classes
~~~~~~~

.. list-table::
   :class: autosummary longtable
   :align: left

   * - :py:obj:`BaseMLP <rofunc.learning.RofuncRL.models.base_models.BaseMLP>`
     -

API
~~~

.. py:class:: BaseMLP(cfg: omegaconf.DictConfig, input_dim: int, output_dim: int, cfg_name: str)
   :canonical: rofunc.learning.RofuncRL.models.base_models.BaseMLP

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

   .. py:method:: forward(x: torch.Tensor) -> torch.Tensor
      :canonical: rofunc.learning.RofuncRL.models.base_models.BaseMLP.forward

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.models.base_models.BaseMLP.forward
