Packages

case class StatefulSeq2[T1, T2, T3, S1, S2, M1 <: StatefulModule[T1, T2, S1], M2 <: StatefulModule[T2, T3, S2]](m1: M1 with StatefulModule[T1, T2, S1], m2: M2 with StatefulModule[T2, T3, S2]) extends StatefulModule[T1, T3, (S1, S2)] with Product with Serializable

Linear Supertypes
Serializable, Product, Equals, GenericModule[(T1, (S1, S2)), (T3, (S1, S2))], AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. StatefulSeq2
  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 StatefulSeq2(m1: M1 with StatefulModule[T1, T2, S1], m2: M2 with StatefulModule[T2, T3, S2])

Value Members

  1. def apply[S](a: (T1, (S1, S2)))(implicit arg0: Sc[S]): (T3, (S1, S2))

    Alias of forward

    Alias of forward

    Definition Classes
    GenericModule
  2. def forward[S](x: (T1, (S1, S2)))(implicit arg0: Sc[S]): (T3, (S1, S2))

    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
    StatefulSeq2GenericModule
  3. def forward1[S](x: T1, st: (S1, S2))(implicit arg0: Sc[S]): (T3, (S1, S2))
  4. 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
  5. final def learnableParameters: Long

    Returns the total number of optimizable parameters.

    Returns the total number of optimizable parameters.

    Definition Classes
    GenericModule
  6. val m1: M1 with StatefulModule[T1, T2, S1]
  7. val m2: M2 with StatefulModule[T2, T3, S2]
  8. 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
  9. def productElementNames: Iterator[String]
    Definition Classes
    Product
  10. def state: Seq[(Constant, PTag)]

    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
    StatefulSeq2GenericModule
  11. final def zeroGrad(): Unit
    Definition Classes
    GenericModule