pulsatrix
Loading...
Searching...
No Matches
pulsatrix::GFlowNetForwardPolicy Member List

This is the complete list of members for pulsatrix::GFlowNetForwardPolicy, including all inherited members.

action_dim() constpulsatrix::GFlowNetForwardPolicyinline
GFlowNetForwardPolicy(Module *policy_network, int64_t action_dim, DeviceBackend *backend, uint32_t seed=42)pulsatrix::GFlowNetForwardPolicy
masked_probs(const Tensor &observation, const std::vector< bool > &valid_actions)pulsatrix::GFlowNetForwardPolicy
sample(const Tensor &observation, const std::vector< bool > &valid_actions)pulsatrix::GFlowNetForwardPolicy