gflownet_trajectory.hpp File Reference

Rolls out one full HyperGrid episode under a GFlowNetForwardPolicy. More...

#include <cstdint>
#include <vector>
#include "pulsatrix/gflownet_forward_policy.hpp"
#include "pulsatrix/hypergrid_env.hpp"
#include "pulsatrix/tensor.hpp"

Include dependency graph for gflownet_trajectory.hpp:

Go to the source code of this file.

Classes
struct	pulsatrix::GFlowNetTrajectory
	One full sampled trajectory: every state the forward policy acted from, the action taken at each, and the trajectory-level quantities Trajectory Balance (and Detailed Balance/SubTB, in later missions) need. More...

Namespaces
namespace	pulsatrix

Functions
GFlowNetTrajectory	pulsatrix::sample_gflownet_trajectory (HyperGridEnv &env, GFlowNetForwardPolicy &forward_policy)
	Samples one full trajectory: resets env, then repeatedly samples a masked action from forward_policy and steps env until termination (an explicit stop or env's own max_steps cap).

Detailed Description

Rolls out one full HyperGrid episode under a GFlowNetForwardPolicy.