Struct relearn::torch::agents::features::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.