Enum juice::weight::FillerType

pub enum FillerType {
    Constant {
        value: f32,
    },
    Glorot {
        input_size: usize,
        output_size: usize,
    },
}

Enum for specifing the type of Filler.

Variants

Constant

Fields

value: f32

The value that will be used to fill the blob.

Fills the weight blob with a constant value (all values are the same).

Glorot

Fields

input_size: usize

Number of input nodes for each output.

output_size: usize

Number of output nodes for each input.

Fills the weight blobs based on the paper:

[Bengio and Glorot 2010]: Understanding the difficulty of training deep feedforward neural networks.

Also known as Xavier filler.

Implementations

impl FillerType

pub fn fill(&self, weight: &mut SharedTensor<f32>)

Uses a filler as specified by this FillerType to fill the values in a SharedTensor

This filling of weights is usually done directly after creation of the weight blob.

pub fn fill_constant(weight: &mut SharedTensor<f32>, value: f32)

Directly use the Constant Filler.

pub fn fill_glorot( weight: &mut SharedTensor<f32>, num_inputs: usize, num_outputs: usize )

Directly use the Glorot Filler.

Trait Implementations

impl Clone for FillerType

fn clone(&self) -> FillerType

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for FillerType

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

Formats the value using the given formatter.

impl Copy for FillerType