ai.djl.training.initializer

Class ConstantInitializer

java.lang.Object

ai.djl.training.initializer.ConstantInitializer

All Implemented Interfaces:

Initializer

public class ConstantInitializer
extends java.lang.Object
implements Initializer

Initializer that generates tensors with constant values.

Field Summary

Fields inherited from interface ai.djl.training.initializer.Initializer

ONES, ZEROS

Constructor Summary

Constructors

Constructor and Description
ConstantInitializer(float value) Creates a Constant Initializer.

Method Summary

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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ConstantInitializer

public ConstantInitializer(float value)

Creates a Constant Initializer.

Parameters:

value - the value to fill

Method Detail

initialize

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

Initializes a single NDArray.

Specified by:

initialize in interface Initializer

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