:py:mod:`rofunc.learning.RofuncRL.state_encoders.graph_encoders`
================================================================

.. py:module:: rofunc.learning.RofuncRL.state_encoders.graph_encoders

.. autodoc2-docstring:: rofunc.learning.RofuncRL.state_encoders.graph_encoders
   :allowtitles:

Module Contents
---------------

Classes
~~~~~~~

.. list-table::
   :class: autosummary longtable
   :align: left

   * - :py:obj:`HomoGraphEncoder <rofunc.learning.RofuncRL.state_encoders.graph_encoders.HomoGraphEncoder>`
     -
   * - :py:obj:`HeteroGraphEncoder <rofunc.learning.RofuncRL.state_encoders.graph_encoders.HeteroGraphEncoder>`
     -

API
~~~

.. py:class:: HomoGraphEncoder(in_dim, hidden_dim)
   :canonical: rofunc.learning.RofuncRL.state_encoders.graph_encoders.HomoGraphEncoder

   Bases: :py:obj:`rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder`

   .. py:method:: forward(g, inputs)
      :canonical: rofunc.learning.RofuncRL.state_encoders.graph_encoders.HomoGraphEncoder.forward

      .. autodoc2-docstring:: rofunc.learning.RofuncRL.state_encoders.graph_encoders.HomoGraphEncoder.forward

.. py:class:: HeteroGraphEncoder()
   :canonical: rofunc.learning.RofuncRL.state_encoders.graph_encoders.HeteroGraphEncoder

   Bases: :py:obj:`rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder`