Class LSTM
-
- All Implemented Interfaces:
Serializable
,Cloneable
,TrainingConfig
public class LSTM extends AbstractLSTM
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
LSTM.Builder
-
Field Summary
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.AbstractLSTM
helperAllowFallback
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseRecurrentLayer
rnnDataFormat, weightInitFnRecurrent
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer
nIn, nOut, timeDistributedFormat
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iUpdater, regularization, regularizationBias, weightInitFn, weightNoise
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer
constraints, iDropout, layerName
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description LayerMemoryReport
getMemoryReport(InputType inputType)
This is a report of the estimated memory consumption for the given layerprotected void
initializeConstraints(Layer.Builder<?> builder)
Initialize the weight constraints.ParamInitializer
initializer()
Layer
instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseRecurrentLayer
getOutputType, getPreProcessorForInputType, setNIn
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer
isPretrainParam
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer
clone, getGradientNormalization, getRegularizationByParam, getUpdaterByParam, resetLayerDefaultConfig
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer
setDataType
-
Methods inherited from class java.lang.Object
equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
Methods inherited from interface org.deeplearning4j.nn.api.TrainingConfig
getGradientNormalizationThreshold, getLayerName
-
-
-
-
Method Detail
-
initializeConstraints
protected void initializeConstraints(Layer.Builder<?> builder)
Description copied from class:Layer
Initialize the weight constraints. Should be called last, in the outer-most constructor- Overrides:
initializeConstraints
in classLayer
-
instantiate
public Layer instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
- Specified by:
instantiate
in classLayer
-
initializer
public ParamInitializer initializer()
- Specified by:
initializer
in classLayer
- Returns:
- The parameter initializer for this model
-
getMemoryReport
public LayerMemoryReport getMemoryReport(InputType inputType)
Description copied from class:Layer
This is a report of the estimated memory consumption for the given layer- Specified by:
getMemoryReport
in classLayer
- Parameters:
inputType
- Input type to the layer. Memory consumption is often a function of the input type- Returns:
- Memory report for the layer
-
-