org.nd4j.linalg.api.ops.impl.layers.recurrent.outputs

Class LSTMLayerOutputs

java.lang.Object

org.nd4j.linalg.api.ops.impl.layers.recurrent.outputs.LSTMLayerOutputs

public class LSTMLayerOutputs
extends Object

The outputs of a LSTM layer (LSTMLayer.

Constructor Summary

Constructors

Constructor and Description
LSTMLayerOutputs(SDVariable[] outputs, LSTMLayerConfig lstmLayerConfig)

Method Summary

Modifier and Type	Method and Description
SDVariable	getLastOutput()
SDVariable	getLastState()
SDVariable	getOutput() Get h, the output of the cell for all time steps.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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()