org.deeplearning4j.nn.api

Interface ParamInitializer

All Known Implementing Classes:

BatchNormalizationParamInitializer, CenterLossParamInitializer, ConvolutionParamInitializer, DefaultParamInitializer, EmptyParamInitializer, GravesBidirectionalLSTMParamInitializer, GravesLSTMParamInitializer, PretrainParamInitializer, VariationalAutoencoderParamInitializer

public interface ParamInitializer

Param initializer for a layer

Method Summary

Modifier and Type	Method and Description
java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray>	getGradientsFromFlattened(NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.INDArray gradientView) Return a map of gradients (in their standard non-flattened representation), taken from the flattened (row vector) gradientView array.
java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray>	init(NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.INDArray paramsView, boolean initializeParams) Initialize the parameters
int	numParams(NeuralNetConfiguration conf)

Method Detail

numParams

int numParams(NeuralNetConfiguration conf)

init

java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray> init(NeuralNetConfiguration conf,
                                                                          org.nd4j.linalg.api.ndarray.INDArray paramsView,
                                                                          boolean initializeParams)

Initialize the parameters

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

java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray> getGradientsFromFlattened(NeuralNetConfiguration conf,
                                                                                               org.nd4j.linalg.api.ndarray.INDArray gradientView)

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 the paramsView does in #init(Map, NeuralNetConfiguration, INDArray); thus the position in the view (and, the array orders) must match those of the parameters

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