Skip to main content

add_gradient_noise

optirs_core::gradient_processing

Function add_gradient_noise 

Source 
pub fn add_gradient_noise<A, D>(
    gradients: &mut Array<A, D>,
    noise_std: A,
    seed: Option<u64>,
) -> &mut Array<A, D>
where
    A: Float + ScalarOperand,
    D: Dimension,
Expand description

Add noise to gradients for regularization

ยงArguments

  • gradients - Gradients to add noise to
  • noise_std - Standard deviation of Gaussian noise to add
  • seed - Optional seed for reproducible results. Some(seed) draws from a deterministically seeded generator, so the same inputs always produce the same noise; None draws from the thread-local generator.