Class SRUCellOutputs
- java.lang.Object
-
- org.nd4j.linalg.api.ops.impl.layers.recurrent.outputs.SRUCellOutputs
-
public class SRUCellOutputs extends Object
-
-
Constructor Summary
Constructors Constructor Description SRUCellOutputs(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.SDVariable
getOutput()
Get h, the output of the cell.SDVariable
getState()
Get c, the state of the cell.
-
-
-
Constructor Detail
-
SRUCellOutputs
public SRUCellOutputs(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].
-
getState
public SDVariable getState()
Get c, the state of the cell. Has shape [batchSize, inSize].
-
-