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

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

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

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

Classes
~~~~~~~

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

   * - :py:obj:`ASEDiscEnc <rofunc.learning.RofuncRL.models.misc_models.ASEDiscEnc>`
     -
   * - :py:obj:`DTrans <rofunc.learning.RofuncRL.models.misc_models.DTrans>`
     -

API
~~~

.. py:class:: ASEDiscEnc(cfg: omegaconf.DictConfig, input_dim: int, enc_output_dim: int, disc_output_dim: int, cfg_name: str)
   :canonical: rofunc.learning.RofuncRL.models.misc_models.ASEDiscEnc

   Bases: :py:obj:`rofunc.learning.RofuncRL.models.base_models.BaseMLP`

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

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.models.misc_models.ASEDiscEnc.forward

   .. py:method:: get_enc(x: torch.Tensor) -> torch.Tensor
      :canonical: rofunc.learning.RofuncRL.models.misc_models.ASEDiscEnc.get_enc

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.models.misc_models.ASEDiscEnc.get_enc

   .. py:method:: get_disc(x: torch.Tensor) -> torch.Tensor
      :canonical: rofunc.learning.RofuncRL.models.misc_models.ASEDiscEnc.get_disc

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.models.misc_models.ASEDiscEnc.get_disc

.. py:class:: DTrans(cfg: omegaconf.DictConfig, observation_space: typing.Optional[typing.Union[int, typing.Tuple[int], gym.Space, gymnasium.Space]], action_space: typing.Optional[typing.Union[int, typing.Tuple[int], gym.Space, gymnasium.Space]], state_encoder: typing.Optional[torch.nn.Module] = EmptyEncoder(), cfg_name='actor')
   :canonical: rofunc.learning.RofuncRL.models.misc_models.DTrans

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

   .. py:method:: forward(states, actions, rewards, returns_to_go, timesteps, attention_mask=None)
      :canonical: rofunc.learning.RofuncRL.models.misc_models.DTrans.forward

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.models.misc_models.DTrans.forward