lamp-core/lamp.nn/StatefulSeq2

StatefulSeq2

case class StatefulSeq2[T1, T2, T3, S1, S2, M1 <: StatefulModule[T1, T2, S1], M2 <: StatefulModule[T2, T3, S2]](m1: M1 & StatefulModule[T1, T2, S1], m2: M2 & StatefulModule[T2, T3, S2]) extends StatefulModule[T1, T3, (S1, S2)]
Companion:
object
trait Serializable
trait Product
trait Equals
trait GenericModule[(T1, (S1, S2)), (T3, (S1, S2))]
class Object
trait Matchable
class Any

Value members

Concrete methods

def forward[S : Sc](x: (T1, (S1, S2))): (T3, (S1, S2))
def forward1[S : Sc](x: T1, st: (S1, S2)): (T3, (S1, S2))
override def state: Seq[(Constant, PTag)]
Definition Classes

Inherited methods

def apply[S : Sc](a: (T1, (S1, S2))): (T3, (S1, S2))

Alias of forward

Alias of forward

Inherited from:
GenericModule
final def gradients(loss: Variable, zeroGrad: Boolean): Seq[Option[STen]]

Computes the gradient of loss with respect to the parameters.

Computes the gradient of loss with respect to the parameters.

Inherited from:
GenericModule
final def learnableParameters: Long

Returns the total number of optimizable parameters.

Returns the total number of optimizable parameters.

Inherited from:
GenericModule
final def parameters: Seq[(Constant, PTag)]

Returns the state variables which need gradient computation.

Returns the state variables which need gradient computation.

Inherited from:
GenericModule
def productElementNames: Iterator[String]
Inherited from:
Product
def productIterator: Iterator[Any]
Inherited from:
Product
final def zeroGrad(): Unit
Inherited from:
GenericModule