Struct BatchBackpropScratch 

pub struct BatchBackpropScratch { /* private fields */ }

Reusable buffers for Mlp::backward_batch.

This stores two work buffers sized for the maximum layer dimension.

Implementations

impl BatchBackpropScratch

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

Allocate a backprop scratch buffer suitable for mlp and batch_size.

Trait Implementations

impl Clone for BatchBackpropScratch

fn clone(&self) -> BatchBackpropScratch

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BatchBackpropScratch

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

Formats the value using the given formatter.