Package org.nd4j.weightinit
Interface WeightInitScheme
-
- All Known Implementing Classes:
BaseWeightInitScheme
,ConstantInitScheme
,DistributionInitScheme
,IdentityInitScheme
,LecunUniformInitScheme
,NDArraySupplierInitScheme
,OneInitScheme
,ReluInitScheme
,ReluUniformInitScheme
,SigmoidUniformInitScheme
,UniformInitScheme
,VarScalingNormalFanAvgInitScheme
,VarScalingNormalFanInInitScheme
,VarScalingNormalFanOutInitScheme
,VarScalingNormalUniformFanInInitScheme
,VarScalingNormalUniformFanOutInitScheme
,VarScalingUniformFanAvgInitScheme
,XavierFanInInitScheme
,XavierInitScheme
,XavierUniformInitScheme
,ZeroInitScheme
public interface WeightInitScheme
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description INDArray
create(long[] shape, INDArray paramsView)
Create the arrayINDArray
create(DataType dataType, long... shape)
Create the arraychar
order()
The order of the weight initWeightInit
type()
The type of the weight init
-
-
-
Method Detail
-
create
INDArray create(long[] shape, INDArray paramsView)
Create the array- Parameters:
shape
- the shape of the arrayparamsView
- the parameters view- Returns:
- the created array
-
create
INDArray create(DataType dataType, long... shape)
Create the array- Parameters:
shape
- the shape of the array- Returns:
- the created array
-
order
char order()
The order of the weight init- Returns:
-
type
WeightInit type()
The type of the weight init- Returns:
-
-