public class LSTM extends AbstractLSTM
Modifier and Type | Class and Description |
---|---|
static class |
LSTM.Builder |
distRecurrent, weightInitRecurrent
nIn, nOut
activationFn, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, iUpdater, l1, l1Bias, l2, l2Bias, weightInit, weightNoise
constraints, iDropout, layerName
Modifier and Type | Method and Description |
---|---|
LayerMemoryReport |
getMemoryReport(InputType inputType)
This is a report of the estimated memory consumption for the given layer
|
protected void |
initializeConstraints(Layer.Builder<?> builder)
Initialize the weight constraints.
|
ParamInitializer |
initializer() |
Layer |
instantiate(NeuralNetConfiguration conf,
Collection<TrainingListener> trainingListeners,
int layerIndex,
org.nd4j.linalg.api.ndarray.INDArray layerParamsView,
boolean initializeParams) |
getL1ByParam, getL2ByParam
getOutputType, getPreProcessorForInputType, setNIn
isPretrainParam
clone, getUpdaterByParam, resetLayerDefaultConfig
protected void initializeConstraints(Layer.Builder<?> builder)
Layer
initializeConstraints
in class Layer
public Layer instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, org.nd4j.linalg.api.ndarray.INDArray layerParamsView, boolean initializeParams)
instantiate
in class Layer
public ParamInitializer initializer()
initializer
in class Layer
public LayerMemoryReport getMemoryReport(InputType inputType)
Layer
getMemoryReport
in class Layer
inputType
- Input type to the layer. Memory consumption is often a function of the input typeCopyright © 2018. All rights reserved.