Package ai.djl.training.initializer
Interface Initializer
-
- All Known Implementing Classes:
ConstantInitializer
,NormalInitializer
,TruncatedNormalInitializer
,UniformInitializer
,XavierInitializer
public interface Initializer
-
-
Field Summary
Fields Modifier and Type Field Description static Initializer
ONES
static Initializer
ZEROS
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description NDArray
initialize(NDManager manager, Shape shape, DataType dataType)
Initializes a singleNDArray
.
-
-
-
Field Detail
-
ZEROS
static final Initializer ZEROS
-
ONES
static final Initializer ONES
-
-