public class NDArraySupplierInitScheme extends Object implements WeightInitScheme
| Modifier and Type | Class and Description |
|---|---|
static interface |
NDArraySupplierInitScheme.NDArraySupplier
A simple
facade |
| Constructor and Description |
|---|
NDArraySupplierInitScheme(INDArray arr) |
| Modifier and Type | Method and Description |
|---|---|
INDArray |
create(DataType dataType,
long[] shape)
Create the array
|
INDArray |
create(long[] shape,
INDArray paramsView)
Create the array
|
char |
order()
The order of the weight init
|
WeightInit |
type()
The type of the weight init
|
public NDArraySupplierInitScheme(INDArray arr)
public INDArray create(long[] shape, INDArray paramsView)
WeightInitSchemecreate in interface WeightInitSchemeshape - the shape of the arrayparamsView - the parameters viewpublic INDArray create(DataType dataType, long[] shape)
WeightInitSchemecreate in interface WeightInitSchemeshape - the shape of the arraypublic char order()
WeightInitSchemeorder in interface WeightInitSchemepublic WeightInit type()
WeightInitSchemetype in interface WeightInitSchemeCopyright © 2021. All rights reserved.