| Package | Description |
|---|---|
| org.deeplearning4j.nn.layers.recurrent |
| Modifier and Type | Field and Description |
|---|---|
protected FwdPassReturn |
LSTM.cachedFwdPass |
protected FwdPassReturn |
GravesLSTM.cachedFwdPass |
protected FwdPassReturn |
GravesBidirectionalLSTM.cachedPassBackward |
protected FwdPassReturn |
GravesBidirectionalLSTM.cachedPassForward |
| Modifier and Type | Method and Description |
|---|---|
FwdPassReturn |
LSTMHelper.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) |
static FwdPassReturn |
LSTMHelpers.activateHelper(BaseLayer 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 originalInputWeights,
org.nd4j.linalg.api.ndarray.INDArray biases,
boolean training,
org.nd4j.linalg.api.ndarray.INDArray originalPrevOutputActivations,
org.nd4j.linalg.api.ndarray.INDArray originalPrevMemCellState,
boolean forBackprop,
boolean forwards,
String inputWeightKey,
org.nd4j.linalg.api.ndarray.INDArray maskArray,
boolean hasPeepholeConnections,
LSTMHelper helper,
CacheMode cacheMode,
LayerWorkspaceMgr workspaceMgr)
Returns FwdPassReturn object with activations/INDArrays.
|
| Modifier and Type | Method and Description |
|---|---|
org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> |
LSTMHelper.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) |
static org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> |
LSTMHelpers.backpropGradientHelper(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,
LSTMHelper helper,
LayerWorkspaceMgr workspaceMgr) |
Copyright © 2018. All rights reserved.