Enum NormalizationConfig 

#[non_exhaustive]
pub enum NormalizationConfig {
    Batch(BatchNormConfig),
    Group(GroupNormConfig),
    Instance(InstanceNormConfig),
    Layer(LayerNormConfig),
    Rms(RmsNormConfig),
}

[‘Normalization’] Configuration.

The enum is non-exhaustive to prepare for future additions.

Can be used as a generic configuration for normalization layers:

• Construct a config with arbitrary input features (we suggest 0).
• Clone and match that config to the target input layer, using the NormalizationConfig::with_num_features() method.

Variants (Non-exhaustive)

Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.

Batch(BatchNormConfig)

[‘BatchNorm’] Configuration.

Group(GroupNormConfig)

[‘GroupNorm’] Configuration.

Instance(InstanceNormConfig)

[‘InstanceNorm’] Configuration.

Layer(LayerNormConfig)

[‘LayerNorm’] Configuration.

Rms(RmsNormConfig)

[‘RmsNorm’] Configuration.

Implementations

impl NormalizationConfig

pub fn init<B>(&self, device: &<B as Backend>::Device) -> Normalization<B>

where B: Backend,

Initialize a [‘Norm’] layer.

pub fn with_num_features(self, num_features: usize) -> NormalizationConfig

Set the number of features.

pub fn num_features(&self) -> usize

Get the number of features.

Trait Implementations

impl Clone for NormalizationConfig

fn clone(&self) -> NormalizationConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Config for NormalizationConfig

fn save<P>(&self, file: P) -> Result<(), Error>

where P: AsRef<Path>,

Saves the configuration to a file.

fn load<P>(file: P) -> Result<Self, ConfigError>

where P: AsRef<Path>,

Loads the configuration from a file.

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for NormalizationConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for NormalizationConfig

fn deserialize<D>( deserializer: D, ) -> Result<NormalizationConfig, <D as Deserializer<'de>>::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for NormalizationConfig

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

Formats the value using the given formatter.

impl From<BatchNormConfig> for NormalizationConfig

fn from(config: BatchNormConfig) -> NormalizationConfig

Converts to this type from the input type.

impl From<GroupNormConfig> for NormalizationConfig

fn from(config: GroupNormConfig) -> NormalizationConfig

Converts to this type from the input type.

impl From<InstanceNormConfig> for NormalizationConfig

fn from(config: InstanceNormConfig) -> NormalizationConfig

Converts to this type from the input type.

impl From<LayerNormConfig> for NormalizationConfig

fn from(config: LayerNormConfig) -> NormalizationConfig

Converts to this type from the input type.

impl From<RmsNormConfig> for NormalizationConfig

fn from(config: RmsNormConfig) -> NormalizationConfig

Converts to this type from the input type.

impl Serialize for NormalizationConfig

fn serialize<S>( &self, serializer: S, ) -> Result<<S as Serializer>::Ok, <S as Serializer>::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.