lamp-core/lamp.nn/Seq2SeqWithAttention

Seq2SeqWithAttention

case class Seq2SeqWithAttention[S0, S1, M0 <: Module, M1 <: StatefulModule2[Variable, Variable, S0, S1], M2 <: StatefulModule[Variable, Variable, S1]](destinationEmbedding: M0 & Module, encoder: M1 & StatefulModule2[Variable, Variable, S0, S1], decoder: M2 & StatefulModule[Variable, Variable, S1], padToken: Long)(stateToKey: S1 => Variable) extends StatefulModule2[(Variable, Variable), Variable, S0, S1]
Companion:
object
trait Serializable
trait Product
trait Equals
trait GenericModule[((Variable, Variable), S0), (Variable, S1)]
class Object
trait Matchable
class Any

Value members

Concrete methods

override def forward[S : Sc](x: ((Variable, Variable), S0)): (Variable, S1)
Definition Classes
override def state: Seq[(Constant, PTag)]
Definition Classes

Inherited methods

def apply[S : Sc](a: ((Variable, Variable), S0)): (Variable, S1)

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

Concrete fields

val stateToKey: S1 => Variable