public class ConvolutionParamInitializer extends Object implements ParamInitializer
| Modifier and Type | Field and Description |
|---|---|
static String |
BIAS_KEY |
static String |
WEIGHT_KEY |
| Constructor and Description |
|---|
ConvolutionParamInitializer() |
| Modifier and Type | Method and Description |
|---|---|
protected org.nd4j.linalg.api.ndarray.INDArray |
createBias(NeuralNetConfiguration conf) |
protected org.nd4j.linalg.api.ndarray.INDArray |
createWeightMatrix(NeuralNetConfiguration conf) |
void |
init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params,
NeuralNetConfiguration conf)
Initialize the parameters
|
void |
init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params,
NeuralNetConfiguration conf,
org.canova.api.conf.Configuration extraConf)
Initialization via extra parameters where necessary
|
public static final String WEIGHT_KEY
public static final String BIAS_KEY
public void init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params, NeuralNetConfiguration conf)
ParamInitializerinit in interface ParamInitializerparams - the parameters to initializeconf - the configurationpublic void init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params, NeuralNetConfiguration conf, org.canova.api.conf.Configuration extraConf)
ParamInitializerinit in interface ParamInitializerparams - the params to configureconf - the configuration to useextraConf - an extra configuration for extensionsprotected org.nd4j.linalg.api.ndarray.INDArray createBias(NeuralNetConfiguration conf)
protected org.nd4j.linalg.api.ndarray.INDArray createWeightMatrix(NeuralNetConfiguration conf)
Copyright © 2016. All Rights Reserved.