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

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 INDArray	initWeights(double fanIn, double fanOut, int[] shape, WeightInit initScheme, Distribution dist, char order, INDArray paramView) Deprecated.
static INDArray	initWeights(double fanIn, double fanOut, int[] shape, WeightInit initScheme, Distribution dist, INDArray paramView) Deprecated.
static INDArray	initWeights(double fanIn, double fanOut, long[] shape, WeightInit initScheme, Distribution dist, char order, INDArray paramView)
static INDArray	initWeights(double fanIn, double fanOut, long[] shape, WeightInit initScheme, Distribution dist, INDArray paramView) Initializes a matrix with the given weight initialization scheme.
static INDArray	initWeights(int[] shape, float min, float max)
static INDArray	reshapeWeights(int[] shape, INDArray paramsView) Reshape the parameters view, without modifying the paramsView array values.
static INDArray	reshapeWeights(int[] shape, INDArray paramsView, char flatteningOrder) Reshape the parameters view, without modifying the paramsView array values.
static INDArray	reshapeWeights(long[] shape, INDArray paramsView) Reshape the parameters view, without modifying the paramsView array values.
static INDArray	reshapeWeights(long[] shape, 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 INDArray initWeights(int[] shape,
                                   float min,
                                   float max)

initWeights

@Deprecated
public static INDArray initWeights(double fanIn,
                                               double fanOut,
                                               int[] shape,
                                               WeightInit initScheme,
                                               Distribution dist,
                                               INDArray paramView)

Deprecated.

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 INDArray initWeights(double fanIn,
                                   double fanOut,
                                   long[] shape,
                                   WeightInit initScheme,
                                   Distribution dist,
                                   INDArray paramView)

Initializes a matrix with the given weight initialization scheme. Note: Defaults to fortran ('f') order arrays for the weights. Use #initWeights(long[], 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

@Deprecated
public static INDArray initWeights(double fanIn,
                                               double fanOut,
                                               int[] shape,
                                               WeightInit initScheme,
                                               Distribution dist,
                                               char order,
                                               INDArray paramView)

Deprecated.

initWeights

public static INDArray initWeights(double fanIn,
                                   double fanOut,
                                   long[] shape,
                                   WeightInit initScheme,
                                   Distribution dist,
                                   char order,
                                   INDArray paramView)

reshapeWeights

public static INDArray reshapeWeights(int[] shape,
                                      INDArray paramsView)

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

Parameters:

shape - Shape to reshape

paramsView - Parameters array view

reshapeWeights

public static INDArray reshapeWeights(long[] shape,
                                      INDArray paramsView)

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

Parameters:

shape - Shape to reshape

paramsView - Parameters array view

reshapeWeights

public static INDArray reshapeWeights(int[] shape,
                                      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

reshapeWeights

public static INDArray reshapeWeights(long[] shape,
                                      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