Struct GroupNormLayer 

pub struct GroupNormLayer { /* private fields */ }

Group normalization: divides channels into groups and normalizes within each group.

Implementations

impl GroupNormLayer

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

pub fn num_groups(&self) -> usize

pub fn num_channels(&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>

Forward inference on NHWC input [N, H, W, C].

Trait Implementations

impl Clone for GroupNormLayer

fn clone(&self) -> GroupNormLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GroupNormLayer

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

Formats the value using the given formatter.