Struct LayerNormLayer 

pub struct LayerNormLayer { /* private fields */ }

Layer normalization over the last dimension.

Implementations

impl LayerNormLayer

pub fn new( graph: &mut Graph, normalized_shape: usize, eps: f32, ) -> Result<Self, ModelError>

pub fn normalized_shape(&self) -> usize

pub fn gamma_node(&self) -> NodeId

pub fn beta_node(&self) -> NodeId

pub fn forward( &self, graph: &mut Graph, input: NodeId, ) -> Result<NodeId, ModelError>

pub fn forward_inference( &self, graph: &Graph, input: &Tensor, ) -> Result<Tensor, ModelError>

Trait Implementations

impl Clone for LayerNormLayer

fn clone(&self) -> LayerNormLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LayerNormLayer

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

Formats the value using the given formatter.