Class SRULayerOutputs
- java.lang.Object
-
- org.nd4j.linalg.api.ops.impl.layers.recurrent.outputs.SRULayerOutputs
-
public class SRULayerOutputs extends Object
-
-
Constructor Summary
Constructors Constructor Description SRULayerOutputs(SDVariable[] outputs)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description List<SDVariable>getAllOutputs()Get all outputs returned by the cell.SDVariablegetLastOutput()Get y, the output of the cell, for the last time step.SDVariablegetLastState()Get c, the state of the cell, for the last time step.SDVariablegetOutput()Get h, the output of the cell.SDVariablegetState()Get c, the state of the cell.
-
-
-
Constructor Detail
-
SRULayerOutputs
public SRULayerOutputs(SDVariable[] outputs)
-
-
Method Detail
-
getAllOutputs
public List<SDVariable> getAllOutputs()
Get all outputs returned by the cell.
-
getOutput
public SDVariable getOutput()
Get h, the output of the cell. Has shape [batchSize, inSize, timeSeriesLength].
-
getState
public SDVariable getState()
Get c, the state of the cell. Has shape [batchSize, inSize, timeSeriesLength].
-
getLastOutput
public SDVariable getLastOutput()
Get y, the output of the cell, for the last time step. Has shape [batchSize, inSize].
-
getLastState
public SDVariable getLastState()
Get c, the state of the cell, for the last time step. Has shape [batchSize, inSize].
-
-