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

Class SRUCellOutputs

java.lang.Object

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

public class SRUCellOutputs
extends Object

The outputs of a GRU cell (GRUCell.

Constructor Summary

Constructors

Constructor and Description
SRUCellOutputs(SDVariable[] outputs)

Method Summary

Modifier and Type	Method and 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.

Methods inherited from class java.lang.Object

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

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].