org.deeplearning4j.nn.weights

Class WeightInitVarScalingUniformFanOut

java.lang.Object

org.deeplearning4j.nn.weights.WeightInitVarScalingUniformFanOut

All Implemented Interfaces:

Serializable, IWeightInit

public class WeightInitVarScalingUniformFanOut
extends Object
implements IWeightInit

Uniform U[-a,a] with a=3.0/(fanOut)
If a scale is provided, a = 3.0 * scale / fanOut

Author:

Adam Gibson

See Also:

Serialized Form

Field Summary

Fields inherited from interface org.deeplearning4j.nn.weights.IWeightInit

DEFAULT_WEIGHT_INIT_ORDER

Constructor Summary

Constructors

Constructor and Description
WeightInitVarScalingUniformFanOut(Double scale)

Method Summary

Modifier and Type	Method and Description
INDArray	init(double fanIn, double fanOut, long[] shape, char order, INDArray paramView) Initialize parameters in the given view.

Methods inherited from class java.lang.Object

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

Constructor Detail

WeightInitVarScalingUniformFanOut

public WeightInitVarScalingUniformFanOut(Double scale)

Method Detail

init

public INDArray init(double fanIn,
                     double fanOut,
                     long[] shape,
                     char order,
                     INDArray paramView)

Description copied from interface: IWeightInit

Initialize parameters in the given view. Double values are used for fanIn and fanOut as some layers (convolution with stride) results in a non-integer number which may be truncated to zero in certain configurations

Specified by:

init in interface IWeightInit

Parameters:

fanIn - Number of input parameters

fanOut - Number of output parameters

shape - Desired shape of array (users shall assume paramView has this shape after method has finished)

order - Order of array, e.g. Fortran ('f') or C ('c')

paramView - View of parameters to initialize (and reshape)