ai.djl.training.initializer

Interface Initializer

All Known Implementing Classes:

ConstantInitializer, NormalInitializer, TruncatedNormalInitializer, UniformInitializer, XavierInitializer

public interface Initializer

An interface representing an initialization method.

Used to initialize the NDArray parameters stored within a Block.

See Also:

The D2L chapter on numerical stability and initialization

Field Summary

Fields

Modifier and Type	Field and Description
static Initializer	ONES
static Initializer	ZEROS

Method Summary

Modifier and Type	Method and Description
NDArray	initialize(NDManager manager, Shape shape, DataType dataType) Initializes a single NDArray.

Field Detail

ZEROS

static final Initializer ZEROS

ONES

static final Initializer ONES

Method Detail

initialize

NDArray initialize(NDManager manager,
                   Shape shape,
                   DataType dataType)

Initializes a single NDArray.

Parameters:

manager - the NDManager to create the new NDArray in

shape - the Shape for the new NDArray

dataType - the DataType for the new NDArray

Returns:

the NDArray initialized with the manager and shape