Struct burn_core::nn::BatchNormRecord

pub struct BatchNormRecord<B: Backend, const D: usize> {
    pub gamma: <Param<Tensor<B, 1>> as Module<B>>::Record,
    pub beta: <Param<Tensor<B, 1>> as Module<B>>::Record,
    pub running_mean: <RunningState<Tensor<B, 1>> as Module<B>>::Record,
    pub running_var: <RunningState<Tensor<B, 1>> as Module<B>>::Record,
    pub momentum: <f64 as Module<B>>::Record,
    pub epsilon: <f64 as Module<B>>::Record,
}

Fields

gamma: <Param<Tensor<B, 1>> as Module<B>>::Record

beta: <Param<Tensor<B, 1>> as Module<B>>::Record

running_mean: <RunningState<Tensor<B, 1>> as Module<B>>::Record

running_var: <RunningState<Tensor<B, 1>> as Module<B>>::Record

momentum: <f64 as Module<B>>::Record

epsilon: <f64 as Module<B>>::Record

Trait Implementations

impl<B: Clone + Backend, const D: usize> Clone for BatchNormRecord<B, D>

fn clone(&self) -> BatchNormRecord<B, D>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<B: Debug + Backend, const D: usize> Debug for BatchNormRecord<B, D>

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

Formats the value using the given formatter.

impl<B: Backend, const D: usize> Record for BatchNormRecord<B, D>

type Item<S: PrecisionSettings> = BatchNormRecordItem<B, D, S>

fn into_item<S: PrecisionSettings>(self) -> Self::Item<S>

Convert the current record into the corresponding item that follows the given settings.

fn from_item<S: PrecisionSettings>(item: Self::Item<S>) -> Self

Convert the given item into a record.