Package org.nd4j.weightinit.impl

Class VarScalingUniformFanAvgInitScheme

java.lang.Object

org.nd4j.weightinit.BaseWeightInitScheme

org.nd4j.weightinit.impl.VarScalingUniformFanAvgInitScheme

All Implemented Interfaces:

WeightInitScheme

public class VarScalingUniformFanAvgInitScheme
extends BaseWeightInitScheme

Initialize the weight to: range = = 3.0 / Math.sqrt((fanIn + fanOut) / 2) U(-range,range)

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor	Description
VarScalingUniformFanAvgInitScheme(char order, double fanIn, double fanOut)

Method Summary

Modifier and Type	Method	Description
INDArray	doCreate(DataType dataType, long[] shape, INDArray paramsView)
WeightInit	type()	The type of the weight init

Methods inherited from class org.nd4j.weightinit.BaseWeightInitScheme

create, create, handleParamsView, order

Methods inherited from class java.lang.Object

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

Constructor Detail

VarScalingUniformFanAvgInitScheme

public VarScalingUniformFanAvgInitScheme(char order,
                                         double fanIn,
                                         double fanOut)

Method Detail

doCreate

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

Specified by:

doCreate in class BaseWeightInitScheme

type

public WeightInit type()

Description copied from interface: WeightInitScheme

The type of the weight init

Returns: