gfn.samplers.trajectories_sampler

Module Contents

Classes

TrajectoriesSampler

Attributes

ActionsTensor
DonesTensor
LogProbsTensor
StatesTensor

gfn.samplers.trajectories_sampler.ActionsTensor

gfn.samplers.trajectories_sampler.DonesTensor

gfn.samplers.trajectories_sampler.LogProbsTensor

gfn.samplers.trajectories_sampler.StatesTensor

class gfn.samplers.trajectories_sampler.TrajectoriesSampler(env, actions_sampler)

Parameters

• env (gfn.envs.Env) –

• actions_sampler (gfn.samplers.actions_samplers.ActionsSampler) –

sample(n_trajectories)

Parameters

n_trajectories (int) –

Return type

gfn.containers.Trajectories

sample_trajectories(states=None, n_trajectories=None)

Parameters

• states (Optional[gfn.containers.States]) –

• n_trajectories (Optional[int]) –

Return type

gfn.containers.Trajectories