org.deeplearning4j.nn.weights

Class WeightInitSigmoidUniform

java.lang.Object

org.deeplearning4j.nn.weights.WeightInitSigmoidUniform

All Implemented Interfaces:

Serializable, IWeightInit

public class WeightInitSigmoidUniform
extends Object
implements IWeightInit

A version of WeightInitXavierUniform for sigmoid activation functions. U(-r,r) with r=4sqrt(6/(fanIn + 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
WeightInitSigmoidUniform()

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

WeightInitSigmoidUniform

public WeightInitSigmoidUniform()

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)