Struct BatchNorm

pub struct BatchNorm { /* private fields */ }

Implementations

impl BatchNorm

pub fn new( num_features: usize, running_mean: Tensor, running_var: Tensor, weight: Tensor, bias: Tensor, eps: f64, ) -> Result<Self>

pub fn new_no_bias( num_features: usize, running_mean: Tensor, running_var: Tensor, eps: f64, ) -> Result<Self>

pub fn new_with_momentum( num_features: usize, running_mean: Tensor, running_var: Tensor, weight: Tensor, bias: Tensor, eps: f64, momentum: f64, ) -> Result<Self>

pub fn new_no_bias_with_momentum( num_features: usize, running_mean: Tensor, running_var: Tensor, eps: f64, momentum: f64, ) -> Result<Self>

pub fn running_mean(&self) -> &Tensor

pub fn running_var(&self) -> &Tensor

pub fn eps(&self) -> f64

pub fn weight_and_bias(&self) -> Option<(&Tensor, &Tensor)>

pub fn momentum(&self) -> f64

pub fn forward_train(&self, x: &Tensor) -> Result<Tensor>

Trait Implementations

impl Clone for BatchNorm

fn clone(&self) -> BatchNorm

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for BatchNorm

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

Formats the value using the given formatter.

impl ModuleT for BatchNorm

fn forward_t(&self, x: &Tensor, train: bool) -> Result<Tensor>