RealmOA Library

oa::ml::nn::BatchNorm2d

Trainable FP32 BatchNorm over channels of NCHW matrices.

StRust public API

Declaration

pub struct BatchNorm2d { /* private fields */ }
Public Rust declaration.

Public Methods

pub fn forward(&self, input: &Matrix) -> Result<Matrix>
pub const fn num_features(&self) -> usize
pub const fn epsilon(&self) -> f32
pub const fn momentum(&self) -> f32
pub fn weight(&self) -> Parameter
pub fn bias(&self) -> Parameter
pub fn running_mean(&self) -> Matrix
pub fn running_variance(&self) -> Matrix
pub fn parameters(&self) -> [Parameter; 2]
fn forward(&self, input: &Matrix) -> Result<Matrix>
fn registry(&self) -> &ModuleRegistry
fn parameters(&self) -> Vec<Parameter>
fn named_parameters(&self) -> Vec<NamedParameter>
fn buffers(&self) -> Vec<Matrix>
fn named_buffers(&self) -> Vec<NamedBuffer>
fn all_parameters(&self) -> Result<Vec<Parameter>>
fn all_named_parameters(&self) -> Result<Vec<NamedParameter>>
fn all_named_buffers(&self) -> Result<Vec<NamedBuffer>>
fn all_named_state_u32(&self) -> Result<Vec<NamedStateU32>>
fn num_parameters(&self) -> Result<usize>
fn train(&self, training: bool)
fn eval(&self)
fn is_training(&self) -> bool
fn scoped_eval(&self) -> ScopedEval<'_>

Associated Functions

pub fn new(engine: &Engine, num_features: usize) -> Result<Self>
pub fn with_options( engine: &Engine, num_features: usize, epsilon: f32, momentum: f32, ) -> Result<Self>
pub fn from_matrices( weight: Matrix, bias: Matrix, running_mean: Matrix, running_variance: Matrix, epsilon: f32, momentum: f32, ) -> Result<Self>

Public Method Documentation

pub fn forward( &self, input: &Matrix ) -> Result<Matrix>
No public source comment is attached to this declaration.
pub const fn num_features( &self ) -> usize
No public source comment is attached to this declaration.
pub const fn epsilon( &self ) -> f32
No public source comment is attached to this declaration.
pub const fn momentum( &self ) -> f32
No public source comment is attached to this declaration.
pub fn weight( &self ) -> Parameter
No public source comment is attached to this declaration.
pub fn bias( &self ) -> Parameter
No public source comment is attached to this declaration.
pub fn running_mean( &self ) -> Matrix
No public source comment is attached to this declaration.
pub fn running_variance( &self ) -> Matrix
No public source comment is attached to this declaration.
pub fn parameters( &self ) -> [Parameter; 2]
No public source comment is attached to this declaration.
fn forward( &self, input: &Matrix ) -> Result<Matrix>
No public source comment is attached to this declaration.
fn registry( &self ) -> &ModuleRegistry
No public source comment is attached to this declaration.
fn parameters( &self ) -> Vec<Parameter>
No public source comment is attached to this declaration.
fn named_parameters( &self ) -> Vec<NamedParameter>
No public source comment is attached to this declaration.
fn buffers( &self ) -> Vec<Matrix>
No public source comment is attached to this declaration.
fn named_buffers( &self ) -> Vec<NamedBuffer>
No public source comment is attached to this declaration.
fn all_parameters( &self ) -> Result<Vec<Parameter>>
No public source comment is attached to this declaration.
fn all_named_parameters( &self ) -> Result<Vec<NamedParameter>>
No public source comment is attached to this declaration.
fn all_named_buffers( &self ) -> Result<Vec<NamedBuffer>>
No public source comment is attached to this declaration.
fn all_named_state_u32( &self ) -> Result<Vec<NamedStateU32>>
No public source comment is attached to this declaration.
fn num_parameters( &self ) -> Result<usize>
No public source comment is attached to this declaration.
fn train( &self, training: bool )
No public source comment is attached to this declaration.
fn eval( &self )
No public source comment is attached to this declaration.
fn is_training( &self ) -> bool
No public source comment is attached to this declaration.
fn scoped_eval( &self ) -> ScopedEval<'_>
No public source comment is attached to this declaration.
pub fn new( engine: &Engine, num_features: usize ) -> Result<Self>
No public source comment is attached to this declaration.
pub fn with_options( engine: &Engine, num_features: usize, epsilon: f32, momentum: f32 ) -> Result<Self>
No public source comment is attached to this declaration.
pub fn from_matrices( weight: Matrix, bias: Matrix, running_mean: Matrix, running_variance: Matrix, epsilon: f32, momentum: f32 ) -> Result<Self>
No public source comment is attached to this declaration.