org.deeplearning4j.nn.params

Class Convolution3DParamInitializer

java.lang.Object

org.deeplearning4j.nn.params.ConvolutionParamInitializer

org.deeplearning4j.nn.params.Convolution3DParamInitializer

All Implemented Interfaces:

ParamInitializer

public class Convolution3DParamInitializer
extends ConvolutionParamInitializer

Initialize 3D convolution parameters.

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static String	BIAS_KEY
static String	WEIGHT_KEY

Constructor Summary

Constructors

Constructor and Description
Convolution3DParamInitializer()

Method Summary

Modifier and Type	Method and Description
protected INDArray	createWeightMatrix(NeuralNetConfiguration conf, INDArray weightView, boolean initializeParams)
Map<String,INDArray>	getGradientsFromFlattened(NeuralNetConfiguration conf, INDArray gradientView) Return a map of gradients (in their standard non-flattened representation), taken from the flattened (row vector) gradientView array.
static Convolution3DParamInitializer	getInstance()
Map<String,INDArray>	init(NeuralNetConfiguration conf, INDArray paramsView, boolean initializeParams) Initialize the parameters
long	numParams(Layer l)
long	numParams(NeuralNetConfiguration conf)

Methods inherited from class org.deeplearning4j.nn.params.ConvolutionParamInitializer

biasKeys, createBias, isBiasParam, isWeightParam, paramKeys, weightKeys

Methods inherited from class java.lang.Object

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

Field Detail

WEIGHT_KEY

public static final String WEIGHT_KEY

See Also:

Constant Field Values

BIAS_KEY

public static final String BIAS_KEY

See Also:

Constant Field Values

Constructor Detail

Convolution3DParamInitializer

public Convolution3DParamInitializer()

Method Detail

getInstance

public static Convolution3DParamInitializer getInstance()

numParams

public long numParams(NeuralNetConfiguration conf)

Specified by:

numParams in interface ParamInitializer

Overrides:

numParams in class ConvolutionParamInitializer

numParams

public long numParams(Layer l)

Specified by:

numParams in interface ParamInitializer

Overrides:

numParams in class ConvolutionParamInitializer

init

public Map<String,INDArray> init(NeuralNetConfiguration conf,
                                 INDArray paramsView,
                                 boolean initializeParams)

Description copied from interface: ParamInitializer

Initialize the parameters

Specified by:

init in interface ParamInitializer

Overrides:

init in class ConvolutionParamInitializer

Parameters:

conf - the configuration

paramsView - a view of the full network (backprop) parameters

initializeParams - if true: initialize the parameters according to the configuration. If false: don't modify the values in the paramsView array (but do select out the appropriate subset, reshape etc as required)

Returns:

Map of parameters keyed by type (view of the 'paramsView' array)

getGradientsFromFlattened

public Map<String,INDArray> getGradientsFromFlattened(NeuralNetConfiguration conf,
                                                      INDArray gradientView)

Description copied from interface: ParamInitializer

Return a map of gradients (in their standard non-flattened representation), taken from the flattened (row vector) gradientView array. The idea is that operates in exactly the same way as the paramsView does in #init(Map, NeuralNetConfiguration, INDArray); thus the position in the view (and, the array orders) must match those of the parameters

Specified by:

getGradientsFromFlattened in interface ParamInitializer

Overrides:

getGradientsFromFlattened in class ConvolutionParamInitializer

Parameters:

conf - Configuration

gradientView - The flattened gradients array, as a view of the larger array

Returns:

A map containing an array by parameter type, that is a view of the full network gradients array

createWeightMatrix

protected INDArray createWeightMatrix(NeuralNetConfiguration conf,
                                      INDArray weightView,
                                      boolean initializeParams)

Overrides:

createWeightMatrix in class ConvolutionParamInitializer