gflownet.policy.state_flow
==========================

.. py:module:: gflownet.policy.state_flow


Classes
-------

.. autoapisummary::

   gflownet.policy.state_flow.StateFlow


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

.. py:class:: StateFlow(config, env, device, float_precision, base=None)

   Bases: :py:obj:`gflownet.policy.base.ModelBase`


   Takes state in the policy format and predicts its flow (a scalar)


   .. py:attribute:: output_dim
      :value: 1



   .. py:method:: instantiate()


   .. py:method:: __call__(states)

      Returns a tensor of the state flows of the shape (batch_size, )