Packages

package attention

Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. attention
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Type Members

  1. abstract class Attention[T, State, StateShape] extends AnyRef
  2. 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]])]
  3. 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
  4. class BahdanauAttention[T] extends SimpleAttention[T]
  5. class LuongAttention[T] extends SimpleAttention[T]
  6. abstract class SimpleAttention[T] extends Attention[T, Output[T], core.Shape]

Value Members

  1. object Attention
  2. object AttentionWrapperCell
  3. object BahdanauAttention
  4. object LuongAttention

Inherited from AnyRef

Inherited from Any

Ungrouped