Struct GradientMask 

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

Gradient masking and freezing utilities

Allows selective gradient updates by masking certain parameters

Implementations

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

pub fn new(mask: Array<bool, D>) -> Self

Create a new gradient mask

Arguments

• mask - Boolean mask indicating which parameters to update

pub fn freeze_all(shape: D) -> Self

Create a mask that freezes all parameters

pub fn update_all(shape: D) -> Self

Create a mask that updates all parameters

pub fn with_lr_multipliers(self, multipliers: Array<A, D>) -> Self

Set learning rate multipliers for different parameters

pub fn apply_mask<'a>( &self, gradients: &'a mut Array<A, D>, ) -> &'a mut Array<A, D>

Apply the mask to gradients

Arguments

• gradients - Gradients to mask

Returns

Masked gradients where frozen parameters have zero gradients

pub fn freeze_indices(&mut self, indices: &[usize]) -> Result<()>

Freeze specific parameters by indices

pub fn unfreeze_indices(&mut self, indices: &[usize]) -> Result<()>

Unfreeze specific parameters by indices

pub fn num_frozen(&self) -> usize

Get the number of frozen parameters

pub fn num_active(&self) -> usize

Get the number of active (unfrozen) parameters

Trait Implementations

impl<A: Clone + Float, D: Clone + Dimension> Clone for GradientMask<A, D>

fn clone(&self) -> GradientMask<A, D>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

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

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

Formats the value using the given formatter.