Neural Network Components
=========================

Low-level neural network layers and attention mechanisms.

Attention Mechanisms
--------------------

.. autoclass:: spatialvi.nn.SpatialAttention
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: spatialvi.nn.CrossAttention
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: spatialvi.nn.NeighborAttention
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: spatialvi.nn.GATLayer
   :members:
   :undoc-members:
   :show-inheritance:

Encoders
--------

.. automodule:: spatialvi.nn._encoders
   :members:
   :undoc-members:
   :show-inheritance:

Decoders
--------

.. automodule:: spatialvi.nn._decoders
   :members:
   :undoc-members:
   :show-inheritance:

Layers
------

.. automodule:: spatialvi.nn._layers
   :members:
   :undoc-members:
   :show-inheritance: