org.deeplearning4j.nn

Class WeightInitUtil

java.lang.Object

org.deeplearning4j.nn.WeightInitUtil

public class WeightInitUtil
extends Object

Weight initialization utility

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
WeightInitUtil()

Method Summary

Modifier and Type	Method and Description
static org.nd4j.linalg.api.ndarray.INDArray	initWeights(int[] shape, float min, float max)
static org.nd4j.linalg.api.ndarray.INDArray	initWeights(int nIn, int nOut, WeightInit initScheme, org.nd4j.linalg.api.activation.ActivationFunction act, org.apache.commons.math3.distribution.RealDistribution dist) Initializes a matrix with the given weight initialization scheme

Methods inherited from class java.lang.Object

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

Constructor Detail

WeightInitUtil

public WeightInitUtil()

Method Detail

initWeights

public static org.nd4j.linalg.api.ndarray.INDArray initWeights(int[] shape,
                                                               float min,
                                                               float max)

initWeights

public static org.nd4j.linalg.api.ndarray.INDArray initWeights(int nIn,
                                                               int nOut,
                                                               WeightInit initScheme,
                                                               org.nd4j.linalg.api.activation.ActivationFunction act,
                                                               org.apache.commons.math3.distribution.RealDistribution dist)

Initializes a matrix with the given weight initialization scheme

Parameters:

nIn - the number of rows in the matrix

nOut - the number of columns in the matrix

initScheme - the scheme to use

Returns:

a matrix of the specified dimensions with the specified distribution based on the initialization scheme