public abstract class BaseWeightInitScheme extends Object implements WeightInitScheme
WeightInitScheme
This handles boilerplate like delegating to the parameters view.Constructor and Description |
---|
BaseWeightInitScheme()
Initialize with c weight ordering by default
|
BaseWeightInitScheme(char order) |
Modifier and Type | Method and Description |
---|---|
INDArray |
create(DataType dataType,
long... shape)
Create the array
|
INDArray |
create(long[] shape,
INDArray paramsView)
Create the array
|
abstract INDArray |
doCreate(DataType dataType,
long[] shape,
INDArray paramsView) |
protected INDArray |
handleParamsView(INDArray outputArray,
INDArray paramView) |
char |
order()
The order of the weight init
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
type
public BaseWeightInitScheme()
public BaseWeightInitScheme(char order)
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
Copyright © 2020. All rights reserved.