public class EmptyParamInitializer extends Object implements ParamInitializer
Constructor and Description |
---|
EmptyParamInitializer() |
Modifier and Type | Method and Description |
---|---|
List<String> |
biasKeys(Layer layer)
Bias parameter keys given the layer configuration
|
Map<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.
|
static EmptyParamInitializer |
getInstance() |
Map<String,org.nd4j.linalg.api.ndarray.INDArray> |
init(NeuralNetConfiguration conf,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams)
Initialize the parameters
|
boolean |
isBiasParam(Layer layer,
String key)
Is the specified parameter a bias?
|
boolean |
isWeightParam(Layer layer,
String key)
Is the specified parameter a weight?
|
int |
numParams(Layer layer) |
int |
numParams(NeuralNetConfiguration conf) |
List<String> |
paramKeys(Layer layer)
Get a list of all parameter keys given the layer configuration
|
List<String> |
weightKeys(Layer layer)
Weight parameter keys given the layer configuration
|
public static EmptyParamInitializer getInstance()
public int numParams(NeuralNetConfiguration conf)
numParams
in interface ParamInitializer
public int numParams(Layer layer)
numParams
in interface ParamInitializer
public List<String> paramKeys(Layer layer)
ParamInitializer
paramKeys
in interface ParamInitializer
layer
- Layerpublic List<String> weightKeys(Layer layer)
ParamInitializer
weightKeys
in interface ParamInitializer
layer
- Layerpublic List<String> biasKeys(Layer layer)
ParamInitializer
biasKeys
in interface ParamInitializer
layer
- Layerpublic boolean isWeightParam(Layer layer, String key)
ParamInitializer
isWeightParam
in interface ParamInitializer
layer
- Layerkey
- Key to checkpublic boolean isBiasParam(Layer layer, String key)
ParamInitializer
isBiasParam
in interface ParamInitializer
layer
- Layerkey
- Key to checkpublic Map<String,org.nd4j.linalg.api.ndarray.INDArray> init(NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.INDArray paramsView, boolean initializeParams)
ParamInitializer
init
in interface ParamInitializer
conf
- the configurationparamsView
- a view of the full network (backprop) parametersinitializeParams
- 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)public Map<String,org.nd4j.linalg.api.ndarray.INDArray> getGradientsFromFlattened(NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.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
conf
- ConfigurationgradientView
- The flattened gradients array, as a view of the larger arrayCopyright © 2018. All rights reserved.