attention

Type Members

1. abstract class Attention[AS, ASS] extends AnyRef

   

2. class AttentionWrapperCell[S, SS, AS, ASS] extends RNNCell[Output, core.Shape, AttentionWrapperState[S, SS, Seq[AS], Seq[ASS]], (SS, core.Shape, core.Shape, Seq[core.Shape], Seq[core.Shape], Seq[ASS])]

   

3. case class AttentionWrapperState[S, SS, AS, ASS](cellState: S, time: Output, attention: Output, alignments: Seq[Output], alignmentsHistory: Seq[TensorArray], attentionState: AS)(implicit evS: Aux[S, SS], evAS: Aux[AS, ASS]) extends Product with Serializable

   

4. class BahdanauAttention extends SimpleAttention

   

5. class LuongAttention extends SimpleAttention

   

6. abstract class SimpleAttention extends Attention[Output, core.Shape]