public class VariationalAutoencoderParamInitializer extends DefaultParamInitializer
| Modifier and Type | Field and Description |
|---|---|
static String |
BIAS_KEY_SUFFIX |
static String |
DECODER_PREFIX |
static String |
ENCODER_PREFIX |
static String |
PXZ_B
Key for bias parameters connecting the last decoder layer and p(data|z) (according to whatever
ReconstructionDistribution is set for the VAE) |
static String |
PXZ_PREFIX |
static String |
PXZ_W
Key for weight parameters connecting the last decoder layer and p(data|z) (according to whatever
ReconstructionDistribution is set for the VAE) |
static String |
PZX_LOGSTD2_B
Key for bias parameters for log(sigma^2) in p(z|data)
|
static String |
PZX_LOGSTD2_PREFIX |
static String |
PZX_LOGSTD2_W
Key for weight parameters connecting the last encoder layer and the log(sigma^2) values for p(z|data)
|
static String |
PZX_MEAN_B
Key for bias parameters for the mean values for p(z|data)
|
static String |
PZX_MEAN_PREFIX |
static String |
PZX_MEAN_W
Key for weight parameters connecting the last encoder layer and the mean values for p(z|data)
|
static String |
PZX_PREFIX |
static String |
WEIGHT_KEY_SUFFIX |
BIAS_KEY, GAIN_KEY, WEIGHT_KEY| Constructor and Description |
|---|
VariationalAutoencoderParamInitializer() |
| Modifier and Type | Method and Description |
|---|---|
List<String> |
biasKeys(Layer layer)
Bias parameter keys given the layer configuration
|
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 VariationalAutoencoderParamInitializer |
getInstance() |
Map<String,INDArray> |
init(NeuralNetConfiguration conf,
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?
|
long |
numParams(NeuralNetConfiguration conf) |
List<String> |
paramKeys(Layer l)
Get a list of all parameter keys given the layer configuration
|
List<String> |
weightKeys(Layer layer)
Weight parameter keys given the layer configuration
|
createBias, createBias, createGain, createGain, createWeightMatrix, createWeightMatrix, hasBias, hasLayerNorm, numParamspublic static final String WEIGHT_KEY_SUFFIX
public static final String BIAS_KEY_SUFFIX
public static final String PZX_PREFIX
public static final String PZX_MEAN_PREFIX
public static final String PZX_LOGSTD2_PREFIX
public static final String ENCODER_PREFIX
public static final String DECODER_PREFIX
public static final String PZX_MEAN_W
public static final String PZX_MEAN_B
public static final String PZX_LOGSTD2_W
public static final String PZX_LOGSTD2_B
public static final String PXZ_PREFIX
public static final String PXZ_W
ReconstructionDistribution is set for the VAE)public static final String PXZ_B
ReconstructionDistribution is set for the VAE)public VariationalAutoencoderParamInitializer()
public static VariationalAutoencoderParamInitializer getInstance()
public long numParams(NeuralNetConfiguration conf)
numParams in interface ParamInitializernumParams in class DefaultParamInitializerpublic List<String> paramKeys(Layer l)
ParamInitializerparamKeys in interface ParamInitializerparamKeys in class DefaultParamInitializerl - Layerpublic List<String> weightKeys(Layer layer)
ParamInitializerweightKeys in interface ParamInitializerweightKeys in class DefaultParamInitializerlayer - Layerpublic List<String> biasKeys(Layer layer)
ParamInitializerbiasKeys in interface ParamInitializerbiasKeys in class DefaultParamInitializerlayer - Layerpublic boolean isWeightParam(Layer layer, String key)
ParamInitializerisWeightParam in interface ParamInitializerisWeightParam in class DefaultParamInitializerlayer - Layerkey - Key to checkpublic boolean isBiasParam(Layer layer, String key)
ParamInitializerisBiasParam in interface ParamInitializerisBiasParam in class DefaultParamInitializerlayer - Layerkey - Key to checkpublic Map<String,INDArray> init(NeuralNetConfiguration conf, INDArray paramsView, boolean initializeParams)
ParamInitializerinit in interface ParamInitializerinit in class DefaultParamInitializerconf - 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,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 DefaultParamInitializerconf - ConfigurationgradientView - The flattened gradients array, as a view of the larger arrayCopyright © 2020. All rights reserved.