Struct BatchScratch 

pub struct BatchScratch { /* private fields */ }

Reusable buffers for Mlp::forward_batch.

This stores per-layer outputs for all samples in the batch in flat row-major layout:

• each layer buffer has shape (batch_size, out_dim).

Implementations

impl BatchScratch

pub fn new(mlp: &Mlp, batch_size: usize) -> Self

Allocate a batch scratch buffer suitable for mlp and batch_size.

pub fn batch_size(&self) -> usize

Returns the fixed batch size.

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

Returns the final model output buffer from the last forward_batch call.

Shape: (batch_size * output_dim,).

pub fn output_row(&self, idx: usize) -> &[f32]

Returns the idx-th output row (shape: (output_dim,)).

Panics if idx >= batch_size.

Trait Implementations

impl Clone for BatchScratch

fn clone(&self) -> BatchScratch

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BatchScratch

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

Formats the value using the given formatter.