GRU

case class GRU(weightXh: Constant, weightHh: Constant, weightXr: Constant, weightXz: Constant, weightHr: Constant, weightHz: Constant, biasR: Constant, biasZ: Constant, biasH: Constant) extends StatefulModule[Variable, Variable, Option[Variable]]

Inputs of size (sequence length * batch * in dim) Outputs of size (sequence length * batch * hidden dim)

trait Serializable

trait Product

trait Equals

class Object

trait Matchable

class Any

Value members

Alias of forward

Computes the gradient of loss with respect to the parameters.

Returns the total number of optimizable parameters.

Returns the state variables which need gradient computation.