:py:mod:`gnn_tracking.models.mlp`
=================================

.. py:module:: gnn_tracking.models.mlp


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

Classes
~~~~~~~

.. autoapisummary::

   gnn_tracking.models.mlp.MLP




.. py:class:: MLP(input_size: int, output_size: int, hidden_dim: int | None, L=3, *, bias=True, include_last_activation=False)


   Bases: :py:obj:`torch.nn.Module`, :py:obj:`pytorch_lightning.core.mixins.HyperparametersMixin`

   Multi Layer Perceptron, using ReLu as activation function.

   :param input_size: Input feature dimension
   :param output_size: Output feature dimension
   :param hidden_dim: Feature dimension of the hidden layers. If None: Choose maximum
                      of input/output size
   :param L: Total number of layers (1 initial layer, L-2 hidden layers, 1 output
             layer)
   :param bias: Include bias in linear layer?
   :param include_last_activation: Include activation function for the last layer?

   .. py:method:: reset_parameters()


   .. py:method:: forward(x)