Struct GroupNormBuilder

pub struct GroupNormBuilder {
    pub group_count: i32,
    pub dimensions: i32,
    pub eps: f32,
    pub affine: bool,
    pub pytorch_compatible: bool,
}

Builder for GroupNorm.

Fields

group_count: i32

Number of groups to separate the features into

dimensions: i32

Number of features in the input

eps: f32

Value added to the denominator for numerical stability. Default to GroupNorm::DEFAULT_EPS.

affine: bool

If true, add a trainable weight and bias. Default to GroupNorm::DEFAULT_AFFINE.

pytorch_compatible: bool

If true, perform the group normalization in the same order/grouping as PyTorch. Default to GroupNorm::DEFAULT_PYTORCH_COMPATIBLE.

Implementations

impl GroupNormBuilder

pub fn new(group_count: impl Into<i32>, dimensions: impl Into<i32>) -> Self

Creates a new GroupNormBuilder.

impl GroupNormBuilder

pub fn eps(self, eps: impl Into<f32>) -> Self

Sets the value of [eps].

pub fn affine(self, affine: impl Into<bool>) -> Self

Sets the value of [affine].

pub fn pytorch_compatible(self, pytorch_compatible: impl Into<bool>) -> Self

Sets the value of [pytorch_compatible].

Trait Implementations

impl Builder<GroupNorm> for GroupNormBuilder

type Error = Exception

Error with building

fn build(self) -> Result<GroupNorm, Self::Error>

Build the type

impl Clone for GroupNormBuilder

fn clone(&self) -> GroupNormBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GroupNormBuilder

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

Formats the value using the given formatter.