Struct LazyHistoryFeatures

pub struct LazyHistoryFeatures<'a, OS: Space + ?Sized, AS: Space + ?Sized, E> { /* private fields */ }

Packed history features with lazy evaluation and caching.

Implementations

impl<'a, OS, AS, E> LazyHistoryFeatures<'a, OS, AS, E>

where OS: Space + ?Sized, AS: Space + ?Sized, E: Sequence,

pub fn new<I>( episodes: I, observation_space: &'a OS, action_space: &'a AS, device: Device, ) -> Self

where I: IntoIterator<Item = E>,

pub fn num_steps(&self) -> usize

pub fn num_episodes(&self) -> usize

pub fn is_empty(&self) -> bool

Trait Implementations

impl<'a, OS: Debug + Space + ?Sized, AS: Debug + Space + ?Sized, E: Debug> Debug for LazyHistoryFeatures<'a, OS, AS, E>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'a, OS, AS, E> HistoryFeatures for LazyHistoryFeatures<'a, OS, AS, E>

where OS: FeatureSpace + ?Sized, AS: ReprSpace<Tensor> + ?Sized, E: Sequence<Item = &'a PartialStep<OS::Element, AS::Element>> + IntoIterator<Item = &'a PartialStep<OS::Element, AS::Element>> + Copy, E::IntoIter: DoubleEndedIterator,

fn observation_features(&self) -> &PackedTensor

Packed observation features. A 2D f32 tensor.

fn extended_observation_features(&self) -> (&PackedTensor, &PackedTensor)

Packed extended observation features. Includes interrupted successor observations.

fn actions(&self) -> &PackedTensor

Packed action values.

fn rewards(&self) -> &PackedTensor

Packed rewards. A 1D f32 tensor.

fn device(&self) -> Device

Device on which tensors will be placed.