.. _nn:

torch_geopooling.nn
====================

.. automodule:: torch_geopooling.nn
.. currentmodule:: torch_geopooling.nn

These are basic building blocks for graphs.


Sparse Layers
-------------

.. autosummary::
   :nosignatures:
   :toctree: generated
   :template: class.rst

   torch_geopooling.nn.Embedding2d


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

.. autosummary::
   :nosignatures:
   :toctree: generated
   :template: class.rst

   torch_geopooling.nn.QuadPool2d
   torch_geopooling.nn.AvgQuadPool2d
   torch_geopooling.nn.MaxQuadPool2d


Adaptive Pooling Layers
-----------------------

.. autosummary::
   :nosignatures:
   :toctree: generated
   :template: class.rst

   torch_geopooling.nn.AdaptiveQuadPool2d
   torch_geopooling.nn.AdaptiveMaxQuadPool2d
   torch_geopooling.nn.AdaptiveAvgQuadPool2d