Struct ContinuousReplayBatch 

pub struct ContinuousReplayBatch {
    pub observations: Vec<f32>,
    pub actions: Vec<f32>,
    pub rewards: Vec<f32>,
    pub next_observations: Vec<f32>,
    pub dones: Vec<bool>,
    pub obs_dim: usize,
    pub action_dim: usize,
}

One minibatch sampled from the continuous replay buffer.

All fields are CPU-side primitive vectors; convert to Burn tensors via ContinuousReplayBatch::to_burn_tensors when handing them to the trainer.

Fields

observations: Vec<f32>

Flattened current observations, shape [batch_size * obs_dim].

actions: Vec<f32>

Flattened actions taken, shape [batch_size * action_dim].

rewards: Vec<f32>

Rewards received, length batch_size.

next_observations: Vec<f32>

Flattened next observations, shape [batch_size * obs_dim].

dones: Vec<bool>

Episode-end mask, length batch_size. true means the transition terminated the episode (so the TD target drops the bootstrap term).

obs_dim: usize

Length of one observation slice (so to_burn_tensors can reshape).

action_dim: usize

Length of one action slice (so to_burn_tensors can reshape).

Implementations

impl ContinuousReplayBatch

pub fn len(&self) -> usize

Number of transitions in the batch.

pub fn is_empty(&self) -> bool

true if the batch is empty.

pub fn to_burn_tensors<B: Backend>( &self, device: &B::Device, ) -> ContinuousReplayBurnTensors<B>

Stack the batch into Burn tensors on device.

Returns a named ContinuousReplayBurnTensors struct so trainers can pattern-match named fields rather than positional tuple elements.

Shapes (all on device):

• observations: [batch, obs_dim], f32
• actions: [batch, action_dim], f32
• rewards: [batch], f32
• next_observations: [batch, obs_dim], f32
• dones: [batch], f32 (0.0 / 1.0; the TD-target formula (1 - done) consumes it directly as a float mask).

Trait Implementations

impl Clone for ContinuousReplayBatch

fn clone(&self) -> ContinuousReplayBatch

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ContinuousReplayBatch

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

Formats the value using the given formatter.