Package ai.djl.training.initializer
Class NormalInitializer
- java.lang.Object
-
- ai.djl.training.initializer.NormalInitializer
-
- All Implemented Interfaces:
Initializer
public class NormalInitializer extends java.lang.Object implements Initializer
NormalInitializerinitializes weights with random values sampled from a normal distribution with a mean of zero and standard deviation ofsigma. Default standard deviation is 0.01.
-
-
Field Summary
-
Fields inherited from interface ai.djl.training.initializer.Initializer
ONES, ZEROS
-
-
Constructor Summary
Constructors Constructor Description NormalInitializer()Creates an instance ofNormalInitializerwith a default sigma of 0.01.NormalInitializer(float sigma)Creates a Normal initializer.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description NDArrayinitialize(NDManager manager, Shape shape, DataType dataType)Initializes a singleNDArray.
-
-
-
Method Detail
-
initialize
public NDArray initialize(NDManager manager, Shape shape, DataType dataType)
Initializes a singleNDArray.- Specified by:
initializein interfaceInitializer- Parameters:
manager- theNDManagerto create the newNDArrayinshape- theShapefor the new NDArraydataType- theDataTypefor the new NDArray- Returns:
- the
NDArrayinitialized with the manager and shape
-
-