gflownet_forward_policy.hpp File Reference

Masked stochastic categorical policy over a logit-producing Module – P_F for GFlowNet training objectives. More...

#include <cstdint>
#include <vector>
#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/module.hpp"
#include "pulsatrix/tensor.hpp"

Include dependency graph for gflownet_forward_policy.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	pulsatrix::GFlowNetSampledAction
	One sample from GFlowNetForwardPolicy::sample: the chosen action and the log-probability the masked distribution assigned to it. More...
class	pulsatrix::GFlowNetForwardPolicy
	Samples from softmax(mask(policy_network(observation))) – a categorical policy restricted to a caller-supplied set of valid actions at the current state. More...

Namespaces
namespace	pulsatrix

Detailed Description

Masked stochastic categorical policy over a logit-producing Module – P_F for GFlowNet training objectives.