Struct DirectSampler 

pub struct DirectSampler<E: Env, H: DirectSamplerHook<E = E>> { /* private fields */ }

Rollout sampler whose workers write directly to output buffers.

Implementations

impl<E: Env, H: DirectSamplerHook<E = E>> DirectSampler<E, H>

pub fn new(core: DirectSamplerCore<E>, hook: H) -> Self

pub fn build<EB: EnvBuilder<Env = E>>( env_builder: EnvBuilderType<EB>, hook: H, execution_mode: SamplerExecutionMode, ) -> Self

Builds a raw sampler and its environment workers.

pub fn build_from_env_builder( env_builder: Arc<dyn EnvBuilder<Env = E>>, num_envs: usize, hook: H, execution_mode: SamplerExecutionMode, ) -> Result<Self>

where E: 'static,

Builds a homogeneous sampler from a shared environment builder.

Errors

Returns an error if num_envs is zero.

Trait Implementations

impl<E: Env, H: DirectSamplerHook<E = E>> Sampler for DirectSampler<E, H>

type Tensor = <E as Env>::Tensor

Tensor type stored in collected trajectories.

fn reset_all_envs(&mut self) -> Result<()>

Resets all environments managed by the sampler.

fn collect_rollouts<A: Actor<Tensor = Self::Tensor> + Clone>( &mut self, actor: A, ) -> Result<()>

Collects rollout data using the provided actor.

fn trajectory_views(&mut self) -> impl AsRef<[TrajectoryView<'_, Self::Tensor>]>

Creates a view for the agents.