grl.neural_network
=====================

.. currentmodule:: grl.neural_network

.. automodule:: grl.neural_network

ConcatenateLayer
-----------------
.. autoclass:: ConcatenateLayer
    :special-members: __init__
    :members:

MultiLayerPerceptron
---------------------
.. autoclass:: MultiLayerPerceptron
    :special-members: __init__
    :members:

ConcatenateMLP
--------------
.. autoclass:: ConcatenateMLP
    :special-members: __init__
    :members:

TemporalSpatialResidualNet
---------------------------
.. autoclass:: TemporalSpatialResidualNet
    :special-members: __init__
    :members:

DiT
---
.. autoclass:: DiT
    :special-members: __init__
    :members:

DiT1D
------
.. autoclass:: DiT1D
    :special-members: __init__
    :members:

DiT2D
------
.. autoclass:: DiT2D
    :special-members: __init__
    :members:


DiT3D
------
.. autoclass:: DiT3D
    :special-members: __init__
    :members: