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

Class LSTMCellOutputs

java.lang.Object

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

public class LSTMCellOutputs
extends Object

Constructor Summary

Constructors

Constructor and Description
LSTMCellOutputs(SDVariable[] outputs)

Method Summary

Modifier and Type	Method and Description
List<SDVariable>	getAllOutputs() Get all outputs returned by the cell.
SDVariable	getOutput() Get y, the output of the cell.
SDVariable	getState() Get c, the cell's state.

Methods inherited from class java.lang.Object

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

Constructor Detail

LSTMCellOutputs

public LSTMCellOutputs(SDVariable[] outputs)

Method Detail

getAllOutputs

public List<SDVariable> getAllOutputs()

Get all outputs returned by the cell.

getOutput

public SDVariable getOutput()

Get y, the output of the cell. Has shape [batchSize, numUnits].

getState

public SDVariable getState()

Get c, the cell's state. Has shape [batchSize, numUnits].