org.nd4j.weightinit

Class BaseWeightInitScheme

java.lang.Object

org.nd4j.weightinit.BaseWeightInitScheme

All Implemented Interfaces:

WeightInitScheme

Direct Known Subclasses:

ConstantInitScheme, DistributionInitScheme, IdentityInitScheme, LecunUniformInitScheme, OneInitScheme, ReluInitScheme, ReluUniformInitScheme, SigmoidUniformInitScheme, UniformInitScheme, VarScalingNormalFanAvgInitScheme, VarScalingNormalFanInInitScheme, VarScalingNormalFanOutInitScheme, VarScalingNormalUniformFanInInitScheme, VarScalingNormalUniformFanOutInitScheme, VarScalingUniformFanAvgInitScheme, XavierFanInInitScheme, XavierInitScheme, XavierUniformInitScheme, ZeroInitScheme

public abstract class BaseWeightInitScheme
extends Object
implements WeightInitScheme

Constructor Summary

Constructors

Constructor and Description
BaseWeightInitScheme() Initialize with c weight ordering by default
BaseWeightInitScheme(char order)

Method Summary

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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.nd4j.weightinit.WeightInitScheme

type

Constructor Detail

BaseWeightInitScheme

public BaseWeightInitScheme()

Initialize with c weight ordering by default

BaseWeightInitScheme

public BaseWeightInitScheme(char order)

Method Detail

doCreate

public abstract INDArray doCreate(DataType dataType,
                                  long[] shape,
                                  INDArray paramsView)

create

public INDArray create(long[] shape,
                       INDArray paramsView)

Description copied from interface: WeightInitScheme

Create the array

Specified by:

create in interface WeightInitScheme

Parameters:

shape - the shape of the array

paramsView - 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 interface WeightInitScheme

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 interface WeightInitScheme

Returns:

handleParamsView

protected INDArray handleParamsView(INDArray outputArray,
                                    INDArray paramView)