public class NormalInitializer extends java.lang.Object implements Initializer
NormalInitializer
initializes weights with random values sampled from a normal
distribution with a mean of zero and standard deviation of sigma
. Default standard
deviation is 0.01.ONES, ZEROS
Constructor and Description |
---|
NormalInitializer()
Creates an instance of
NormalInitializer with a default sigma of 0.01. |
NormalInitializer(float sigma)
Creates a Normal initializer.
|
Modifier and Type | Method and Description |
---|---|
NDArray |
initialize(NDManager manager,
Shape shape,
DataType dataType)
Initializes a single
NDArray . |
public NormalInitializer()
NormalInitializer
with a default sigma of 0.01.public NormalInitializer(float sigma)
sigma
- the standard deviation of the normal distributionpublic NDArray initialize(NDManager manager, Shape shape, DataType dataType)
NDArray
.initialize
in interface Initializer
manager
- the NDManager
to create the new NDArray
inshape
- the Shape
for the new NDArraydataType
- the DataType
for the new NDArrayNDArray
initialized with the manager and shape