Trait InPlaceOptimizer 

pub trait InPlaceOptimizer<A: Float + ScalarOperand + Debug, D: Dimension> {
    // Required method
    fn step_inplace(
        &mut self,
        params: &mut Array<A, D>,
        gradients: &Array<A, D>,
    ) -> Result<()>;

    // Provided method
    fn step_list_inplace(
        &mut self,
        params_list: &mut [&mut Array<A, D>],
        gradients_list: &[&Array<A, D>],
    ) -> Result<()> { ... }
}

Trait for in-place parameter updates

Required Methods

fn step_inplace( &mut self, params: &mut Array<A, D>, gradients: &Array<A, D>, ) -> Result<()>

Update parameters in-place using the given gradients

This method modifies the parameters directly rather than returning new arrays, which can significantly reduce memory usage for large models.

Provided Methods

fn step_list_inplace( &mut self, params_list: &mut [&mut Array<A, D>], gradients_list: &[&Array<A, D>], ) -> Result<()>

Update multiple parameter arrays in-place

Implementors

impl<A: Float + ScalarOperand + Debug, D: Dimension + Send + Sync> InPlaceOptimizer<A, D> for InPlaceAdam<A, D>

impl<A: Float + ScalarOperand + Debug, D: Dimension + Send + Sync> InPlaceOptimizer<A, D> for InPlaceSGD<A>