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)
WeightInitScheme
create
in interface WeightInitScheme
shape
- the shape of the arrayparamsView
- the parameters viewpublic INDArray create(DataType dataType, long[] shape)
WeightInitScheme
create
in interface WeightInitScheme
shape
- the shape of the arraypublic char order()
WeightInitScheme
order
in interface WeightInitScheme
public WeightInit type()
WeightInitScheme
type
in interface WeightInitScheme
Copyright © 2022. All rights reserved.