org.deeplearning4j.nn.layers.recurrent

Interface LSTMHelper

All Superinterfaces:

LayerHelper

public interface LSTMHelper
extends LayerHelper

Helper for the recurrent LSTM layer (no peephole connections).

Author:

saudet

Method Summary

Modifier and Type	Method and Description
FwdPassReturn	activate(Layer layer, NeuralNetConfiguration conf, IActivation gateActivationFn, INDArray input, INDArray recurrentWeights, INDArray inputWeights, INDArray biases, boolean training, INDArray prevOutputActivations, INDArray prevMemCellState, boolean forBackprop, boolean forwards, String inputWeightKey, INDArray maskArray, boolean hasPeepholeConnections, LayerWorkspaceMgr workspaceMgr)
Pair<Gradient,INDArray>	backpropGradient(NeuralNetConfiguration conf, IActivation gateActivationFn, INDArray input, INDArray recurrentWeights, INDArray inputWeights, INDArray epsilon, boolean truncatedBPTT, int tbpttBackwardLength, FwdPassReturn fwdPass, boolean forwards, String inputWeightKey, String recurrentWeightKey, String biasWeightKey, Map<String,INDArray> gradientViews, INDArray maskArray, boolean hasPeepholeConnections, LayerWorkspaceMgr workspaceMgr)
boolean	checkSupported(IActivation gateActivationFn, IActivation activationFn, boolean hasPeepholeConnections)

Methods inherited from interface org.deeplearning4j.nn.layers.LayerHelper

helperMemoryUse

Method Detail

checkSupported

boolean checkSupported(IActivation gateActivationFn,
                       IActivation activationFn,
                       boolean hasPeepholeConnections)

backpropGradient

Pair<Gradient,INDArray> backpropGradient(NeuralNetConfiguration conf,
                                         IActivation gateActivationFn,
                                         INDArray input,
                                         INDArray recurrentWeights,
                                         INDArray inputWeights,
                                         INDArray epsilon,
                                         boolean truncatedBPTT,
                                         int tbpttBackwardLength,
                                         FwdPassReturn fwdPass,
                                         boolean forwards,
                                         String inputWeightKey,
                                         String recurrentWeightKey,
                                         String biasWeightKey,
                                         Map<String,INDArray> gradientViews,
                                         INDArray maskArray,
                                         boolean hasPeepholeConnections,
                                         LayerWorkspaceMgr workspaceMgr)

activate

FwdPassReturn activate(Layer layer,
                       NeuralNetConfiguration conf,
                       IActivation gateActivationFn,
                       INDArray input,
                       INDArray recurrentWeights,
                       INDArray inputWeights,
                       INDArray biases,
                       boolean training,
                       INDArray prevOutputActivations,
                       INDArray prevMemCellState,
                       boolean forBackprop,
                       boolean forwards,
                       String inputWeightKey,
                       INDArray maskArray,
                       boolean hasPeepholeConnections,
                       LayerWorkspaceMgr workspaceMgr)