torch_frame.nn
==============

.. contents:: Contents
    :local:

.. currentmodule:: torch_frame.nn

torch_frame.nn.encoder
----------------------

.. currentmodule:: torch_frame.nn.encoder

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

   {% for name in torch_frame.nn.encoder.classes %}
     {{ name }}
   {% endfor %}


torch_frame.nn.encoding
------------------------

.. currentmodule:: torch_frame.nn.encoding

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

   {% for name in torch_frame.nn.encoding.classes %}
     {{ name }}
   {% endfor %}

torch_frame.nn.conv
------------------------

.. currentmodule:: torch_frame.nn.conv

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

   {% for name in torch_frame.nn.conv.classes %}
     {{ name }}
   {% endfor %}

torch_frame.nn.decoder
------------------------

.. currentmodule:: torch_frame.nn.decoder

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

   {% for name in torch_frame.nn.decoder.classes %}
     {{ name }}
   {% endfor %}

torch_frame.nn.models
------------------------

.. currentmodule:: torch_frame.nn.models

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

   {% for name in torch_frame.nn.models.classes %}
     {{ name }}
   {% endfor %}