Struct BatchNorm2d 

pub struct BatchNorm2d<T: Num, D: Device> {
    pub scale: Variable<T, D>,
    pub bias: Variable<T, D>,
    pub mean: Variable<T, D>,
    pub variance: Variable<T, D>,
    /* private fields */
}

Fields

scale: Variable<T, D>

bias: Variable<T, D>

mean: Variable<T, D>

variance: Variable<T, D>

Implementations

impl<T: Num, D: Device> BatchNorm2d<T, D>

pub fn new(channels: usize, momentum: f64) -> Self

pub fn to<Dout: Device>(self) -> BatchNorm2d<T, Dout>

Trait Implementations

impl<T: Num, D: Device> Module<T, D> for BatchNorm2d<T, D>

type Input = Variable<T, D>

type Output = Variable<T, D>

fn call(&self, input: Variable<T, D>) -> Variable<T, D>

impl<T: Num, D: Device> Parameters<T, D> for BatchNorm2d<T, D>

fn weights(&self) -> HashMap<String, Variable<T, D>>

fn biases(&self) -> HashMap<String, Variable<T, D>>

fn parameters(&self) -> HashMap<String, Variable<T, D>>

fn load_parameters(&mut self, parameters: HashMap<String, Variable<T, D>>)