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 interface
NDArraySupplierInitScheme.NDArraySupplier
A simplefacade
-
Constructor Summary
Constructors Constructor Description NDArraySupplierInitScheme(INDArray arr)
-
Method Summary
All Methods Instance Methods Concrete 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
-
-
-
Constructor Detail
-
NDArraySupplierInitScheme
public NDArraySupplierInitScheme(INDArray arr)
-
-
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:
-
type
public WeightInit type()
Description copied from interface:WeightInitScheme
The type of the weight init- Specified by:
type
in interfaceWeightInitScheme
- Returns:
-
-