Struct SpectralNormalization 

pub struct SpectralNormalization<T: FloatBounds> { /* private fields */ }

Spectral normalization for constraining the spectral norm of weight matrices

Implementations

impl<T: FloatBounds + ScalarOperand> SpectralNormalization<T>

pub fn new(power_iterations: usize, eps: T) -> Self

Create a new spectral normalization instance

pub fn normalize_weights(&mut self, weights: &Array2<T>) -> Array2<T>

where T: ScalarOperand + Clone + Debug,

Apply spectral normalization to a weight matrix

pub fn get_spectral_norm(&mut self, weights: &Array2<T>) -> T

where T: ScalarOperand + Clone,

Get the current estimate of the spectral norm

pub fn reset(&mut self)

Reset the cached vectors (useful when weight dimensions change)

Trait Implementations

impl<T: Clone + FloatBounds> Clone for SpectralNormalization<T>

fn clone(&self) -> SpectralNormalization<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Debug + FloatBounds> Debug for SpectralNormalization<T>

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

Formats the value using the given formatter.

impl<T: FloatBounds + ScalarOperand> Default for SpectralNormalization<T>

fn default() -> Self

Returns the “default value” for a type.