Struct border_atari_env::util::test::ObsBatch

pub struct ObsBatch {
    pub n: usize,
    pub m: usize,
    pub buf: Vec<u8>,
}

Consists the observation part of a batch in SimpleReplayBuffer.

Fields

n: usize

The number of samples in the batch.

m: usize

The length of a sample in bytes.

buf: Vec<u8>

The buffer.

Trait Implementations

impl From<BorderAtariObs> for ObsBatch

fn from(obs: Obs) -> Self

Converts to this type from the input type.

impl SubBatch for ObsBatch

fn new(capacity: usize) -> Self

Builds a subbatch with a capacity.

fn push(&mut self, i: usize, data: &Self)

Pushes the samples in data.

fn sample(&self, ixs: &Vec<usize>) -> Self

Takes samples in the batch.