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 INDArraycreate(long[] shape, INDArray paramsView)Create the arrayINDArraycreate(DataType dataType, long... shape)Create the arraycharorder()The order of the weight initWeightInittype()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:
-
-