Package org.nd4j.weightinit.impl
Class DistributionInitScheme
- java.lang.Object
-
- org.nd4j.weightinit.BaseWeightInitScheme
-
- org.nd4j.weightinit.impl.DistributionInitScheme
-
- All Implemented Interfaces:
WeightInitScheme
public class DistributionInitScheme extends BaseWeightInitScheme
Initialize the weights based on a givenDistribution
- Author:
- Adam Gibson
-
-
Constructor Summary
Constructors Constructor Description DistributionInitScheme(char order, Distribution distribution)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description INDArray
doCreate(DataType dataType, long[] shape, INDArray paramsView)
WeightInit
type()
The type of the weight init-
Methods inherited from class org.nd4j.weightinit.BaseWeightInitScheme
create, create, handleParamsView, order
-
-
-
-
Constructor Detail
-
DistributionInitScheme
public DistributionInitScheme(char order, Distribution distribution)
-
-
Method Detail
-
doCreate
public INDArray doCreate(DataType dataType, long[] shape, INDArray paramsView)
- Specified by:
doCreate
in classBaseWeightInitScheme
-
type
public WeightInit type()
Description copied from interface:WeightInitScheme
The type of the weight init- Returns:
-
-