Class GRUCellOutputs
- java.lang.Object
-
- org.nd4j.linalg.api.ops.impl.layers.recurrent.outputs.GRUCellOutputs
-
public class GRUCellOutputs extends Object
-
-
Constructor Summary
Constructors Constructor Description GRUCellOutputs(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.
-
-
-
Constructor Detail
-
GRUCellOutputs
public GRUCellOutputs(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, numUnits].
-
-