public class DeconvolutionParamInitializer extends ConvolutionParamInitializer
BIAS_KEY, WEIGHT_KEY
Constructor and Description |
---|
DeconvolutionParamInitializer() |
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 DeconvolutionParamInitializer |
getInstance() |
biasKeys, createBias, init, isBiasParam, isWeightParam, numParams, numParams, paramKeys, weightKeys
public static DeconvolutionParamInitializer getInstance()
protected INDArray createWeightMatrix(NeuralNetConfiguration conf, INDArray weightView, boolean initializeParams)
createWeightMatrix
in class ConvolutionParamInitializer
public Map<String,INDArray> getGradientsFromFlattened(NeuralNetConfiguration conf, INDArray gradientView)
ParamInitializer
#init(Map, NeuralNetConfiguration, INDArray)
;
thus the position in the view (and, the array orders) must match those of the parametersgetGradientsFromFlattened
in interface ParamInitializer
getGradientsFromFlattened
in class ConvolutionParamInitializer
conf
- ConfigurationgradientView
- The flattened gradients array, as a view of the larger arrayCopyright © 2019. All rights reserved.