package attention
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- attention
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- Protected
Type Members
- abstract class Attention[T, State, StateShape] extends AnyRef
- class AttentionWrapperCell[T, CellState, AttentionState, CellStateShape, AttentionStateShape] extends RNNCell[Output[T], AttentionWrapperState[T, CellState, AttentionState], core.Shape, (CellStateShape, core.Shape, core.Shape, Seq[core.Shape], Seq[core.Shape], Seq[StateShape[AttentionStateShape]])]
- case class AttentionWrapperState[AttentionDataType, CellState, AttentionState](cellState: CellState, time: Output[Int], attention: Output[AttentionDataType], alignments: Seq[Output[AttentionDataType]], alignmentsHistory: Seq[TensorArray[AttentionDataType]], attentionState: Seq[State[AttentionDataType, AttentionState]]) extends Product with Serializable
- class BahdanauAttention[T] extends SimpleAttention[T]
- class LuongAttention[T] extends SimpleAttention[T]
- abstract class SimpleAttention[T] extends Attention[T, Output[T], core.Shape]
Value Members
- object Attention
- object AttentionWrapperCell
- object BahdanauAttention
- object LuongAttention