public class GravesBidirectionalLSTMParamInitializer extends java.lang.Object implements ParamInitializer
Modifier and Type | Field and Description |
---|---|
static java.lang.String |
BIAS_KEY_BACKWARDS |
static java.lang.String |
BIAS_KEY_FORWARDS |
static java.lang.String |
INPUT_WEIGHT_KEY_BACKWARDS |
static java.lang.String |
INPUT_WEIGHT_KEY_FORWARDS |
static java.lang.String |
RECURRENT_WEIGHT_KEY_BACKWARDS |
static java.lang.String |
RECURRENT_WEIGHT_KEY_FORWARDS
Weights for previous time step -> current time step connections
|
Constructor and Description |
---|
GravesBidirectionalLSTMParamInitializer() |
Modifier and Type | Method and Description |
---|---|
java.util.Map<java.lang.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 GravesBidirectionalLSTMParamInitializer |
getInstance() |
java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray> |
init(NeuralNetConfiguration conf,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams)
Initialize the parameters
|
int |
numParams(NeuralNetConfiguration conf) |
public static final java.lang.String RECURRENT_WEIGHT_KEY_FORWARDS
public static final java.lang.String BIAS_KEY_FORWARDS
public static final java.lang.String INPUT_WEIGHT_KEY_FORWARDS
public static final java.lang.String RECURRENT_WEIGHT_KEY_BACKWARDS
public static final java.lang.String BIAS_KEY_BACKWARDS
public static final java.lang.String INPUT_WEIGHT_KEY_BACKWARDS
public GravesBidirectionalLSTMParamInitializer()
public static GravesBidirectionalLSTMParamInitializer getInstance()
public int numParams(NeuralNetConfiguration conf)
numParams
in interface ParamInitializer
public java.util.Map<java.lang.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 java.util.Map<java.lang.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 array