Package org.nd4j.weightinit.impl
Class NDArraySupplierInitScheme
- java.lang.Object
-
- org.nd4j.weightinit.impl.NDArraySupplierInitScheme
-
- All Implemented Interfaces:
WeightInitScheme
public class NDArraySupplierInitScheme extends Object implements WeightInitScheme
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static interfaceNDArraySupplierInitScheme.NDArraySupplierA simplefacade
-
Constructor Summary
Constructors Constructor Description NDArraySupplierInitScheme(INDArray arr)
-
Method Summary
All Methods Instance Methods Concrete 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
-
-
-
Constructor Detail
-
NDArraySupplierInitScheme
public NDArraySupplierInitScheme(INDArray arr)
-
-
Method Detail
-
create
public INDArray create(long[] shape, INDArray paramsView)
Description copied from interface:WeightInitSchemeCreate the array- Specified by:
createin 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:WeightInitSchemeCreate the array- Specified by:
createin interfaceWeightInitSchemeshape- the shape of the array- Returns:
- the created array
-
order
public char order()
Description copied from interface:WeightInitSchemeThe order of the weight init- Specified by:
orderin interfaceWeightInitScheme- Returns:
-
type
public WeightInit type()
Description copied from interface:WeightInitSchemeThe type of the weight init- Specified by:
typein interfaceWeightInitScheme- Returns:
-
-