torch_geometric.nn
==================

.. contents:: Contents
    :local:

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

.. automodule:: torch_geometric.nn.conv.message_passing
    :members:

.. automodule:: torch_geometric.nn.conv
    :members:
    :undoc-members:
    :exclude-members: message, update, MessagePassing

.. automodule:: torch_geometric.nn.meta
    :members:
    :undoc-members:

Dense Convolutional Layers
--------------------------

.. automodule:: torch_geometric.nn.dense.dense_gcn_conv
    :members:
    :undoc-members:

.. automodule:: torch_geometric.nn.dense.dense_sage_conv
    :members:
    :undoc-members:

.. automodule:: torch_geometric.nn.dense.dense_graph_conv
    :members:
    :undoc-members:

.. automodule:: torch_geometric.nn.dense.dense_gin_conv
    :members:
    :undoc-members:

Normalization Layers
--------------------

.. automodule:: torch_geometric.nn.norm
    :members:
    :undoc-members:

Global Pooling Layers
---------------------

.. automodule:: torch_geometric.nn.glob
    :members:
    :undoc-members:

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

.. automodule:: torch_geometric.nn.pool
    :members:
    :undoc-members:

Dense Pooling Layers
--------------------

.. automodule:: torch_geometric.nn.dense.diff_pool
    :members:
    :undoc-members:

.. automodule:: torch_geometric.nn.dense.mincut_pool
    :members:
    :undoc-members:

Unpooling Layers
----------------

.. automodule:: torch_geometric.nn.unpool
    :members:
    :undoc-members:

Models
------

.. automodule:: torch_geometric.nn.models
    :members:
    :undoc-members:

DataParallel Layers
-------------------

.. automodule:: torch_geometric.nn.data_parallel
    :members:
    :undoc-members: