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.
-
ClassDescriptionInitializer that generates tensors with constant values.An interface representing an initialization method.
NormalInitializerinitializes weights with random values sampled from a normal distribution with a mean of zero and standard deviation ofsigma.Naive implementation of a truncated normal initializer.UniformInitializerinitializes weights with random values uniformly sampled from a given range.XavierInitializeris anInitializerthat performs "Xavier" initialization for parameters.Enum for different types of factor type.Enum for different types of random distributions.