Class GRU¶

Defined in File cells.h

Inheritance Relationships¶

Base Type¶

public marian::rnn::Cell (Class Cell)

Class Documentation¶

class GRU : public marian::rnn::Cell ¶

Base class for a gated recurrent unit (GRU) cell.

Public Functions

GRU(Ptr<ExpressionGraph> graph, Ptr<Options> options)¶

virtual State apply(std::vector<Expr> inputs, State state, Expr mask = nullptr)¶

virtual std::vector<Expr> applyInput(std::vector<Expr> inputs)¶

virtual State applyState(std::vector<Expr> xWs, State state, Expr mask = nullptr)¶

Protected Attributes

std::string prefix_¶

Expr U_¶

Expr W_¶

Expr b_¶

Expr gamma1_¶

Expr gamma2_¶

bool final_¶

bool layerNorm_¶

float dropout_¶

Expr dropMaskX_¶

Expr dropMaskS_¶

Expr fakeInput_¶