org.deeplearning4j.nn.weights

Class WeightInitVarScalingNormalFanIn

java.lang.Object

org.deeplearning4j.nn.weights.WeightInitVarScalingNormalFanIn

All Implemented Interfaces:

Serializable, IWeightInit

public class WeightInitVarScalingNormalFanIn
extends Object
implements IWeightInit

Gaussian distribution with mean 0, variance 1.0/(fanIn)
If a scale is provided, use variance scale/(fanIn) instead

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
WeightInitVarScalingNormalFanIn(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

WeightInitVarScalingNormalFanIn

public WeightInitVarScalingNormalFanIn(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)