Struct GradientAccumulator 

pub struct GradientAccumulator<T> { /* private fields */ }

Gradient accumulator for managing accumulated gradients

Implementations

impl<T: Clone> GradientAccumulator<T>

pub fn new(config: AccumulationConfig) -> Self

pub fn accumulate(&mut self, gradients: Vec<T>)

Add gradients to accumulator

pub fn is_ready(&self) -> bool

Check if ready to step (accumulated enough)

pub fn step(&mut self) -> Vec<T>

Get accumulated gradients and optionally reset

pub fn clear(&mut self)

Clear accumulated gradients

pub fn count(&self) -> usize

Get current accumulation count

pub fn config(&self) -> &AccumulationConfig