org.deeplearning4j.nn.weights

Class WeightInitUtil

java.lang.Object

org.deeplearning4j.nn.weights.WeightInitUtil

public class WeightInitUtil
extends java.lang.Object

Weight initialization utility

Field Summary

Fields

Modifier and Type	Field and Description
static char	DEFAULT_WEIGHT_INIT_ORDER Default order for the arrays created by WeightInitUtil.

Method Summary

Modifier and Type	Method and Description
static org.nd4j.linalg.api.ndarray.INDArray	initWeights(double fanIn, double fanOut, int[] shape, WeightInit initScheme, org.nd4j.linalg.api.rng.distribution.Distribution dist, char order, org.nd4j.linalg.api.ndarray.INDArray paramView)
static org.nd4j.linalg.api.ndarray.INDArray	initWeights(double fanIn, double fanOut, int[] shape, WeightInit initScheme, org.nd4j.linalg.api.rng.distribution.Distribution dist, org.nd4j.linalg.api.ndarray.INDArray paramView) Initializes a matrix with the given weight initialization scheme.
static org.nd4j.linalg.api.ndarray.INDArray	initWeights(int[] shape, float min, float max)
static org.nd4j.linalg.api.ndarray.INDArray	reshapeWeights(int[] shape, org.nd4j.linalg.api.ndarray.INDArray paramsView) Reshape the parameters view, without modifying the paramsView array values.
static org.nd4j.linalg.api.ndarray.INDArray	reshapeWeights(int[] shape, org.nd4j.linalg.api.ndarray.INDArray paramsView, char flatteningOrder) Reshape the parameters view, without modifying the paramsView array values.

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_WEIGHT_INIT_ORDER

public static final char DEFAULT_WEIGHT_INIT_ORDER

Default order for the arrays created by WeightInitUtil.

See Also:

Constant Field Values

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(double fanIn,
                                                               double fanOut,
                                                               int[] shape,
                                                               WeightInit initScheme,
                                                               org.nd4j.linalg.api.rng.distribution.Distribution dist,
                                                               org.nd4j.linalg.api.ndarray.INDArray paramView)

Initializes a matrix with the given weight initialization scheme. Note: Defaults to fortran ('f') order arrays for the weights. Use #initWeights(int[], WeightInit, Distribution, char, INDArray) to control this

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(double fanIn,
                                                               double fanOut,
                                                               int[] shape,
                                                               WeightInit initScheme,
                                                               org.nd4j.linalg.api.rng.distribution.Distribution dist,
                                                               char order,
                                                               org.nd4j.linalg.api.ndarray.INDArray paramView)

reshapeWeights

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

Reshape the parameters view, without modifying the paramsView array values.

Parameters:

shape - Shape to reshape

paramsView - Parameters array view

reshapeWeights

public static org.nd4j.linalg.api.ndarray.INDArray reshapeWeights(int[] shape,
                                                                  org.nd4j.linalg.api.ndarray.INDArray paramsView,
                                                                  char flatteningOrder)

Reshape the parameters view, without modifying the paramsView array values.

Parameters:

shape - Shape to reshape

paramsView - Parameters array view

flatteningOrder - Order in which parameters are flattened/reshaped