Package ai.djl.training.initializer
Class UniformInitializer
- java.lang.Object
-
- ai.djl.training.initializer.UniformInitializer
-
- All Implemented Interfaces:
Initializer
public class UniformInitializer extends java.lang.Object implements Initializer
UniformInitializerinitializes weights with random values uniformly sampled from a given range.
-
-
Field Summary
-
Fields inherited from interface ai.djl.training.initializer.Initializer
ONES, ZEROS
-
-
Constructor Summary
Constructors Constructor Description UniformInitializer()Creates an instance ofUniformInitializerwith a defaultscaleof 0.07.UniformInitializer(float scale)Initializes a uniform initializer.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description NDArrayinitialize(NDManager manager, Shape shape, DataType dataType)Initializes a singleNDArray.
-
-
-
Constructor Detail
-
UniformInitializer
public UniformInitializer()
Creates an instance ofUniformInitializerwith a defaultscaleof 0.07.
-
UniformInitializer
public UniformInitializer(float scale)
Initializes a uniform initializer.- Parameters:
scale- the bound on the range of the generated random values. Values are generated from the range [-`scale`, `scale`]. Default scale is 0.07.
-
-
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
-
-