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
- Alphabetic
- By Inheritance
- StatefulSeq2
- Serializable
- Product
- Equals
- GenericModule
- AnyRef
- Any
- Hide All
- Show All
- Public
- Protected
Instance Constructors
- new StatefulSeq2(m1: M1 with StatefulModule[T1, T2, S1], m2: M2 with StatefulModule[T2, T3, S2])
Value Members
- def apply[S](a: (T1, (S1, S2)))(implicit arg0: Sc[S]): (T3, (S1, S2))
Alias of forward
Alias of forward
- Definition Classes
- GenericModule
- 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 thestate to compute its value.
- Definition Classes
- StatefulSeq2 → GenericModule
- def forward1[S](x: T1, st: (S1, S2))(implicit arg0: Sc[S]): (T3, (S1, S2))
- 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
- final def learnableParameters: Long
Returns the total number of optimizable parameters.
Returns the total number of optimizable parameters.
- Definition Classes
- GenericModule
- val m1: M1 with StatefulModule[T1, T2, S1]
- val m2: M2 with StatefulModule[T2, T3, S2]
- 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
- def productElementNames: Iterator[String]
- Definition Classes
- Product
- 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
- StatefulSeq2 → GenericModule
- final def zeroGrad(): Unit
- Definition Classes
- GenericModule