Class LSTMLayerOutputs
- java.lang.Object
-
- org.nd4j.linalg.api.ops.impl.layers.recurrent.outputs.LSTMLayerOutputs
-
-
Constructor Summary
Constructors Constructor Description LSTMLayerOutputs(SDVariable[] outputs, LSTMLayerConfig lstmLayerConfig)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description SDVariable
getLastOutput()
SDVariable
getLastState()
SDVariable
getOutput()
Get h, the output of the cell for all time steps.
-
-
-
Constructor Detail
-
LSTMLayerOutputs
public LSTMLayerOutputs(SDVariable[] outputs, LSTMLayerConfig lstmLayerConfig)
-
-
Method Detail
-
getOutput
public SDVariable getOutput()
Get h, the output of the cell for all time steps.Shape depends on data format defined in
LSTMLayerConfig
:
for unidirectional: TNS: shape [timeLength, numExamples, inOutSize] - sometimes referred to as "time major"
NST: shape [numExamples, inOutSize, timeLength]
NTS: shape [numExamples, timeLength, inOutSize]
for bidirectional: T2NS: 3 = [timeLength, 2, numExamples, inOutSize] (for ONNX)
-
getLastState
public SDVariable getLastState()
-
getLastOutput
public SDVariable getLastOutput()
-
-