Package ai.djl.training.initializer


package ai.djl.training.initializer
Contains classes for initializing a Block's Parameters.

It contains a main interface Initializer and various initializers that extend it.

  • Class
    Description
    Initializer that generates tensors with constant values.
    An interface representing an initialization method.
    NormalInitializer initializes weights with random values sampled from a normal distribution with a mean of zero and standard deviation of sigma.
    Naive implementation of a truncated normal initializer.
    UniformInitializer initializes weights with random values uniformly sampled from a given range.
    XavierInitializer is an Initializer that performs "Xavier" initialization for parameters.
    Enum for different types of factor type.
    Enum for different types of random distributions.