Struct GradientAccumulator 

pub struct GradientAccumulator<A: Float, D: Dimension> { /* private fields */ }

Gradient accumulator for micro-batch training

Implementations

impl<A: Float + ScalarOperand + Debug, D: Dimension + Send + Sync> GradientAccumulator<A, D>

pub fn new(_targetaccumulations: usize, mode: AccumulationMode) -> Self

Create a new gradient accumulator

pub fn initialize(&mut self, gradients: &[Array<A, D>]) -> Result<()>

Initialize accumulator with gradient shapes

pub fn accumulate(&mut self, gradients: &[Array<A, D>]) -> Result<()>

Accumulate gradients from a micro-batch

pub fn is_ready(&self) -> bool

Check if accumulation is complete

pub fn get_and_reset(&mut self) -> Result<Vec<Array<A, D>>>

Get accumulated gradients and reset accumulator

pub fn reset(&mut self)

Reset accumulator state

pub fn accumulation_count(&self) -> usize

Get current accumulation count

pub fn target_accumulations(&self) -> usize

Get target accumulation count

pub fn set_target_accumulations(&mut self, target: usize)

Set new target accumulation count

pub fn mode(&self) -> AccumulationMode

Get accumulation mode

pub fn set_mode(&mut self, mode: AccumulationMode)

Set accumulation mode

pub fn is_initialized(&self) -> bool

Check if accumulator is initialized

pub fn progress(&self) -> f64

Get current progress as a fraction (0.0 to 1.0)

Trait Implementations

impl<A: Debug + Float, D: Debug + Dimension> Debug for GradientAccumulator<A, D>

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

Formats the value using the given formatter.