org.deeplearning4j.nn.weights

Class WeightInitUtil

java.lang.Object

org.deeplearning4j.nn.weights.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[] shape, WeightInit initScheme, org.nd4j.linalg.api.rng.distribution.Distribution dist) Initializes a matrix with the given weight initialization scheme
static org.nd4j.linalg.api.ndarray.INDArray	initWeights(int nIn, int nOut, WeightInit initScheme, org.nd4j.linalg.api.rng.distribution.Distribution dist) Initializes a matrix with the given weight initialization scheme
static org.nd4j.linalg.api.ndarray.INDArray	normalized(int[] shape, int nIn) Normalized weight init
static org.nd4j.linalg.api.ndarray.INDArray	uniformBasedOnInAndOut(int[] shape, int nIn, int nOut) Generate a random matrix with respect to the number of inputs and outputs.

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

normalized

public static org.nd4j.linalg.api.ndarray.INDArray normalized(int[] shape,
                                                              int nIn)

Normalized weight init

Parameters:

shape - shape

nIn - number of inputs

Returns:

the weights

uniformBasedOnInAndOut

public static org.nd4j.linalg.api.ndarray.INDArray uniformBasedOnInAndOut(int[] shape,
                                                                          int nIn,
                                                                          int nOut)

Generate a random matrix with respect to the number of inputs and outputs. This is a bound uniform distribution with the specified minimum and maximum

Parameters:

shape - the shape of the matrix

nIn - the number of inputs

nOut - the number of outputs

Returns:

INDArray

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[] shape,
                                                               WeightInit initScheme,
                                                               org.nd4j.linalg.api.rng.distribution.Distribution dist)

Initializes a matrix with the given weight initialization scheme

Parameters:

shape - the shape of the matrix

initScheme - the scheme to use

Returns:

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

initWeights

public static org.nd4j.linalg.api.ndarray.INDArray initWeights(int nIn,
                                                               int nOut,
                                                               WeightInit initScheme,
                                                               org.nd4j.linalg.api.rng.distribution.Distribution 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