ML_modules.build_mlp
====================

Back to :doc:`Module page <../modules/ML_modules>`

Summary
-------

Create an multi-layer perceptron (MLP) from the configuration.

Signature
---------

.. code-block:: python

   def build_mlp(input_dim, hidden_dim, output_dim, layers, activation = nn.GELU())

.. list-table:: Input variables
   :header-rows: 1

   * - Name
     - Type
     - Default
     - Description
   * - ``input_dim``
     - -
     - -
     - -
   * - ``hidden_dim``
     - -
     - -
     - -
   * - ``output_dim``
     - -
     - -
     - -
   * - ``layers``
     - -
     - -
     - -
   * - ``activation``
     - -
     - nn.GELU()
     - -

Output variables
----------------

.. list-table::
   :header-rows: 1

   * - Return annotation
     - Docstring type
     - Description
   * - ``None``
     - -
     - -

Docstring
---------

.. code-block:: text

   Create an multi-layer perceptron (MLP) from the configuration.