Struct TrajectoryBuffer 

pub struct TrajectoryBuffer<T: R2lTensor> { /* private fields */ }

Owned, structure-of-arrays storage for one environment trajectory.

Implementations

impl<T: R2lTensor> TrajectoryBuffer<T>

pub fn clear(&mut self)

Removes all stored transitions while retaining allocated capacity.

pub fn push(&mut self, memory: Memory<T>)

Appends one transition to the buffer.

pub fn replace_last_next_state(&mut self, next_state: T)

Replaces the most recently stored next state, if one exists.

pub fn len(&self) -> usize

Returns the number of stored transitions.

pub fn is_empty(&self) -> bool

Returns true when the buffer contains no transitions.

pub fn terminated(&self) -> &[bool]

Returns terminal-state flags.

pub fn truncated(&self) -> &[bool]

Returns truncation flags.

pub fn rewards(&self) -> &[f32]

Returns stored rewards.

pub fn rewards_mut(&mut self) -> &mut [f32]

Returns mutable access to the stored rewards.

pub fn to_trajectory_view(&self) -> TrajectoryView<'_, T>

Borrows the buffer as aligned trajectory slices.

Trait Implementations

impl<T: Clone + R2lTensor> Clone for TrajectoryBuffer<T>

fn clone(&self) -> TrajectoryBuffer<T>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: R2lTensor> Default for TrajectoryBuffer<T>

fn default() -> Self

Returns the “default value” for a type.