public interface WeightInitScheme
BaseWeightInitScheme
to create a new WeightInitScheme
This is needed to handle things like the parameters view.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
|
INDArray create(long[] shape, INDArray paramsView)
shape
- the shape of the arrayparamsView
- the parameters viewINDArray create(DataType dataType, long... shape)
shape
- the shape of the arraychar order()
WeightInit type()
Copyright © 2020. All rights reserved.