border_core::generic_replay_buffer

Struct GenericTransitionBatch

pub struct GenericTransitionBatch<O, A>where
    O: BatchBase,
    A: BatchBase,{
    pub obs: O,
    pub act: A,
    pub next_obs: O,
    pub reward: Vec<f32>,
    pub is_terminated: Vec<i8>,
    pub is_truncated: Vec<i8>,
    pub weight: Option<Vec<f32>>,
    pub ix_sample: Option<Vec<usize>>,
}

Expand description

A generic implementation of TransitionBatch.

Fields§

§obs: O

Observations.

§act: A

Actions.

§next_obs: O

Next observations.

§reward: Vec<f32>

Rewards.

§is_terminated: Vec<i8>

Termination flags.

§is_truncated: Vec<i8>

Truncation flags.

§weight: Option<Vec<f32>>

Priority weights.

§ix_sample: Option<Vec<usize>>

Sample indices.

Implementations§

impl<O, A> GenericTransitionBatch<O, A>
where O: BatchBase, A: BatchBase,

pub fn with_capacity(capacity: usize) -> Self

Creates new batch with the given capacity.

Trait Implementations§

impl<O, A> TransitionBatch for GenericTransitionBatch<O, A>
where O: BatchBase, A: BatchBase,

type ObsBatch = O

A set of observation in a batch.

type ActBatch = A

A set of observation in a batch.

fn unpack( self, ) -> (Self::ObsBatch, Self::ActBatch, Self::ObsBatch, Vec<f32>, Vec<i8>, Vec<i8>, Option<Vec<usize>>, Option<Vec<f32>>)

Unpack the data (o_t, a_t, o_t+n, r_t, is_terminated_t, is_truncated_t). Read more

fn len(&self) -> usize

Returns the number of samples in the batch.

fn obs(&self) -> &Self::ObsBatch

Returns o_t.

Auto Trait Implementations§

impl<O, A> Freeze for GenericTransitionBatch<O, A>
where O: Freeze, A: Freeze,

impl<O, A> RefUnwindSafe for GenericTransitionBatch<O, A>
where O: RefUnwindSafe, A: RefUnwindSafe,

impl<O, A> Send for GenericTransitionBatch<O, A>
where O: Send, A: Send,

impl<O, A> Sync for GenericTransitionBatch<O, A>
where O: Sync, A: Sync,

impl<O, A> Unpin for GenericTransitionBatch<O, A>
where O: Unpin, A: Unpin,

impl<O, A> UnwindSafe for GenericTransitionBatch<O, A>
where O: UnwindSafe, A: UnwindSafe,

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V