Packages

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]] with Product with Serializable

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

Linear Supertypes
Serializable, Product, Equals, GenericModule[(Variable, Option[Variable]), (Variable, Option[Variable])], AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. GRU
  2. Serializable
  3. Product
  4. Equals
  5. GenericModule
  6. AnyRef
  7. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Instance Constructors

  1. new GRU(weightXh: Constant, weightHh: Constant, weightXr: Constant, weightXz: Constant, weightHr: Constant, weightHz: Constant, biasR: Constant, biasZ: Constant, biasH: Constant)

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##: Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. def apply[S](a: (Variable, Option[Variable]))(implicit arg0: Sc[S]): (Variable, Option[Variable])

    Alias of forward

    Alias of forward

    Definition Classes
    GenericModule
  5. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  6. val biasH: Constant
  7. val biasR: Constant
  8. val biasZ: Constant
  9. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.CloneNotSupportedException]) @native() @IntrinsicCandidate()
  10. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  11. def forward[S](a: (Variable, Option[Variable]))(implicit arg0: Sc[S]): (Variable, Some[Variable])

    The implementation of the function.

    The implementation of the function.

    In addition of x it can also use all the state to compute its value.

    Definition Classes
    GRUGenericModule
  12. final def getClass(): Class[_ <: AnyRef]
    Definition Classes
    AnyRef → Any
    Annotations
    @native() @IntrinsicCandidate()
  13. final def gradients(loss: Variable, zeroGrad: Boolean = true): Seq[Option[STen]]

    Computes the gradient of loss with respect to the parameters.

    Computes the gradient of loss with respect to the parameters.

    Definition Classes
    GenericModule
  14. val hiddenSize: Long
  15. def initState: None
  16. val inputSize: Long
  17. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  18. final def learnableParameters: Long

    Returns the total number of optimizable parameters.

    Returns the total number of optimizable parameters.

    Definition Classes
    GenericModule
  19. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  20. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @IntrinsicCandidate()
  21. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @IntrinsicCandidate()
  22. final def parameters: Seq[(Constant, PTag)]

    Returns the state variables which need gradient computation.

    Returns the state variables which need gradient computation.

    Definition Classes
    GenericModule
  23. def productElementNames: Iterator[String]
    Definition Classes
    Product
  24. def state: List[(Constant, LeafTag with Product with Serializable)]

    List of optimizable, or non-optimizable, but stateful parameters

    List of optimizable, or non-optimizable, but stateful parameters

    Stateful means that the state is carried over the repeated forward calls.

    Definition Classes
    GRUGenericModule
  25. final def synchronized[T0](arg0: => T0): T0
    Definition Classes
    AnyRef
  26. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  27. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException]) @native()
  28. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  29. val weightHh: Constant
  30. val weightHr: Constant
  31. val weightHz: Constant
  32. val weightXh: Constant
  33. val weightXr: Constant
  34. val weightXz: Constant
  35. final def zeroGrad(): Unit
    Definition Classes
    GenericModule

Deprecated Value Members

  1. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.Throwable]) @Deprecated
    Deprecated

Inherited from Serializable

Inherited from Product

Inherited from Equals

Inherited from GenericModule[(Variable, Option[Variable]), (Variable, Option[Variable])]

Inherited from AnyRef

Inherited from Any

Ungrouped