fastdev.nn.mlp
==============

.. py:module:: fastdev.nn.mlp






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

.. py:class:: MLP(input_dim: int, output_dim: int, hidden_dims: List[int], activation_layer: Optional[Type[torch.nn.Module]] = nn.ReLU, activation_on_output: bool = False, residual_on_output: bool = False, residual_on_hidden: bool = False, use_normalization: bool = False, normalization_layer: Optional[Type[torch.nn.Module]] = nn.LayerNorm)

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


   A flexible MLP


   .. py:attribute:: dims


   .. py:attribute:: residual_on_hidden
      :value: False



   .. py:attribute:: residual_on_output
      :value: False



   .. py:attribute:: layers


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


.. py:data:: mlp