Struct RMSDiv 

pub struct RMSDiv<E: Float + MatMulImpl, const I: usize> { /* private fields */ }

A layer which divides each feature by the RMS of all features.

Trait Implementations

impl<E: Clone + Float + MatMulImpl, const I: usize> Clone for RMSDiv<E, I>

fn clone(&self) -> RMSDiv<E, I>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<E: Debug + Float + MatMulImpl, const I: usize> Debug for RMSDiv<E, I>

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

Formats the value using the given formatter.

impl<E: Default + Float + MatMulImpl, const I: usize> Default for RMSDiv<E, I>

fn default() -> RMSDiv<E, I>

Returns the “default value” for a type.

impl<E: Float + MatMulImpl, const I: usize> LoadableModule for RMSDiv<E, I>

fn save( &self, _path: String, _dict: &mut HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

Saves the parameters to the given dictionary.

fn load( &mut self, _path: String, _dict: &HashMap<String, Vec<f64>>, ) -> Result<(), LoadSaveError>

Loads the parameters from the given dictionary.

impl<E: Float + MatMulImpl, const I: usize> Module<[E; I]> for RMSDiv<E, I>

type Output = [E; I]

The type that this unit produces given Input.

fn forward(&self, x: &[E; I]) -> Result<Self::Output, Error>

impl<E: Float + MatMulImpl, const I: usize> ResetParams for RMSDiv<E, I>

fn rand_params<RNG: Rng>( &mut self, _rng: &mut RNG, _scale: f32, ) -> Result<(), Error>

Sensibly initializes the learnable parameters of some module.

impl<E: Float + MatMulImpl, const I: usize> RevModule<[E; I]> for RMSDiv<E, I>

type SelfGrads = ()

The type describing gradients with respect to the modules’ own parameters.

fn reverse( &self, inputs: &[E; I], grads_wrt_output: &[E; I], ) -> ([E; I], Self::SelfGrads)

Returns the gradients with respect to the input, and the gradients with respect to any internal parameters.

fn apply( &mut self, _applyer: &mut impl GradApplyer, _updates: Self::SelfGrads, ) -> Result<(), Error>

Applies a gradient update step: adding product of the provided gradients and the scalar to the parameters.

impl<E: Float + MatMulImpl, const I: usize> VisualizableUnit for RMSDiv<E, I>

const KIND: &'static str = "rmsdiv"

type Params = ()

fn params(&self) -> &Self::Params