Package org.nd4j.weightinit
Class BaseWeightInitScheme
- java.lang.Object
-
- org.nd4j.weightinit.BaseWeightInitScheme
-
- All Implemented Interfaces:
WeightInitScheme
- Direct Known Subclasses:
ConstantInitScheme
,DistributionInitScheme
,IdentityInitScheme
,LecunUniformInitScheme
,OneInitScheme
,ReluInitScheme
,ReluUniformInitScheme
,SigmoidUniformInitScheme
,UniformInitScheme
,VarScalingNormalFanAvgInitScheme
,VarScalingNormalFanInInitScheme
,VarScalingNormalFanOutInitScheme
,VarScalingNormalUniformFanInInitScheme
,VarScalingNormalUniformFanOutInitScheme
,VarScalingUniformFanAvgInitScheme
,XavierFanInInitScheme
,XavierInitScheme
,XavierUniformInitScheme
,ZeroInitScheme
public abstract class BaseWeightInitScheme extends Object implements WeightInitScheme
-
-
Constructor Summary
Constructors Constructor Description BaseWeightInitScheme()
Initialize with c weight ordering by defaultBaseWeightInitScheme(char order)
-
Method Summary
All Methods Instance Methods Abstract Methods Concrete Methods Modifier and Type Method Description INDArray
create(long[] shape, INDArray paramsView)
Create the arrayINDArray
create(DataType dataType, long... shape)
Create the arrayabstract INDArray
doCreate(DataType dataType, long[] shape, INDArray paramsView)
protected INDArray
handleParamsView(INDArray outputArray, INDArray paramView)
char
order()
The order of the weight init-
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
Methods inherited from interface org.nd4j.weightinit.WeightInitScheme
type
-
-
-
-
Method Detail
-
create
public INDArray create(long[] shape, INDArray paramsView)
Description copied from interface:WeightInitScheme
Create the array- Specified by:
create
in interfaceWeightInitScheme
- Parameters:
shape
- the shape of the arrayparamsView
- the parameters view- Returns:
- the created array
-
create
public INDArray create(DataType dataType, long... shape)
Description copied from interface:WeightInitScheme
Create the array- Specified by:
create
in interfaceWeightInitScheme
shape
- the shape of the array- Returns:
- the created array
-
order
public char order()
Description copied from interface:WeightInitScheme
The order of the weight init- Specified by:
order
in interfaceWeightInitScheme
- Returns:
-
-