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, weightKeyspublic static DeconvolutionParamInitializer getInstance()
protected INDArray createWeightMatrix(NeuralNetConfiguration conf, INDArray weightView, boolean initializeParams)
createWeightMatrix in class ConvolutionParamInitializerpublic 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 ParamInitializergetGradientsFromFlattened in class ConvolutionParamInitializerconf - ConfigurationgradientView - The flattened gradients array, as a view of the larger arrayCopyright © 2020. All rights reserved.