Package 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	Description
static final Initializer	ONES
static final Initializer	ZEROS

Method Summary

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

Field Details

ZEROS

static final Initializer ZEROS

ONES

static final Initializer ONES

Method Details

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