org.deeplearning4j.nn.layers.recurrent

Interface LSTMHelper

public interface LSTMHelper

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, org.nd4j.linalg.activations.IActivation gateActivationFn, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray recurrentWeights, org.nd4j.linalg.api.ndarray.INDArray inputWeights, org.nd4j.linalg.api.ndarray.INDArray biases, boolean training, org.nd4j.linalg.api.ndarray.INDArray prevOutputActivations, org.nd4j.linalg.api.ndarray.INDArray prevMemCellState, boolean forBackprop, boolean forwards, String inputWeightKey, org.nd4j.linalg.api.ndarray.INDArray maskArray, boolean hasPeepholeConnections, LayerWorkspaceMgr workspaceMgr)
org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray>	backpropGradient(NeuralNetConfiguration conf, org.nd4j.linalg.activations.IActivation gateActivationFn, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray recurrentWeights, org.nd4j.linalg.api.ndarray.INDArray inputWeights, org.nd4j.linalg.api.ndarray.INDArray epsilon, boolean truncatedBPTT, int tbpttBackwardLength, FwdPassReturn fwdPass, boolean forwards, String inputWeightKey, String recurrentWeightKey, String biasWeightKey, Map<String,org.nd4j.linalg.api.ndarray.INDArray> gradientViews, org.nd4j.linalg.api.ndarray.INDArray maskArray, boolean hasPeepholeConnections, LayerWorkspaceMgr workspaceMgr)
boolean	checkSupported(org.nd4j.linalg.activations.IActivation gateActivationFn, org.nd4j.linalg.activations.IActivation activationFn, boolean hasPeepholeConnections)

Method Detail

checkSupported

boolean checkSupported(org.nd4j.linalg.activations.IActivation gateActivationFn,
                       org.nd4j.linalg.activations.IActivation activationFn,
                       boolean hasPeepholeConnections)

backpropGradient

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

activate

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