haiku_geometric.nn
==================

.. contents:: Contents
    :local:

Convolutional Layers
--------------------

.. currentmodule:: haiku_geometric.nn.conv
.. autosummary::
   :nosignatures:
   {% for cls in haiku_geometric.nn.conv.classes %}
     {{ cls }}
   {% endfor %}

.. automodule:: haiku_geometric.nn.conv
   :members:
   :special-members:
   :undoc-members:
   :exclude-members: __abstractmethods__, __module__, __repr__

Pooling Layers
--------------

.. currentmodule:: haiku_geometric.nn.pool
.. autosummary::
   :nosignatures:
   {% for cls in haiku_geometric.nn.pool.classes %}
     {{ cls }}
   {% endfor %}

.. automodule:: haiku_geometric.nn.pool
   :members:
   :special-members:
   :undoc-members:
   :exclude-members: __abstractmethods__, __module__, __repr__

Dense Layers
------------

.. currentmodule:: haiku_geometric.nn.dense
.. autosummary::
   :nosignatures:
   {% for cls in haiku_geometric.nn.dense.classes %}
     {{ cls }}
   {% endfor %}

.. automodule:: haiku_geometric.nn.dense
   :members:
   :undoc-members:
   :special-members:
   :exclude-members: __abstractmethods__, __module__, __repr__

Agregation Operators
--------------------

.. currentmodule:: haiku_geometric.nn.aggr
.. autosummary::
   :nosignatures:
   {% for cls in haiku_geometric.nn.aggr.classes %}
     {{ cls }}
   {% endfor %}

.. automodule:: haiku_geometric.nn.aggr
   :members:
   :undoc-members:

Attention Layers
----------------

.. currentmodule:: haiku_geometric.nn.attention
.. autosummary::
   :nosignatures:
   {% for cls in haiku_geometric.nn.attention.classes %}
     {{ cls }}
   {% endfor %}

.. automodule:: haiku_geometric.nn.attention
   :members:
   :undoc-members:
   :special-members:
   :exclude-members: __abstractmethods__, __module__, __repr__