flint.nn
========

.. automodule:: flint.nn.modules


Module
---------------------------------

.. autoclass:: Module
    :members:
    :undoc-members:
    :show-inheritance:


Containers
---------------------------------

.. autoclass:: Sequential
    :show-inheritance:


Activations
-----------------------------------

.. autoclass:: ReLU
   :show-inheritance:

.. autoclass:: LeakyReLU
   :show-inheritance:

.. autoclass:: Sigmoid
   :show-inheritance:

.. autoclass:: Tanh
   :show-inheritance:

.. autoclass:: GELU
   :show-inheritance:


Loss Functions
----------------------------

.. autoclass:: Loss
   :show-inheritance:

.. autoclass:: BCELoss
   :show-inheritance:

.. autoclass:: MSELoss
   :show-inheritance:

.. autoclass:: NllLoss
   :show-inheritance:

.. autoclass:: CrossEntropyLoss
   :show-inheritance:


Convolution
----------------------------

.. autoclass:: Conv1d
   :show-inheritance:

.. autoclass:: Conv2d
   :show-inheritance:


Linear
------------------------------

.. autoclass:: Linear
   :show-inheritance:

.. autoclass:: Identity
   :show-inheritance:


Pooling
----------------------------

.. autoclass:: MaxPool1d
   :show-inheritance:

.. autoclass:: MaxPool2d
   :show-inheritance:


Unfold
----------------------------
.. autoclass:: Unfold
   :show-inheritance:


Dropout
----------------------------

.. autoclass:: Dropout
    :show-inheritance:


Flatten
----------------------------

.. autoclass:: Flatten
    :show-inheritance: