MultiHeadCrossAttention class

Combines multiple cross-attention heads to run in parallel.

Inheritance

• Object
• Module
• MultiHeadCrossAttention

Constructors

MultiHeadCrossAttention(int numHeads, int decoderEmbedSize, int encoderEmbedSize)

Properties

hashCode → int

The hash code for this object.

no setter, inherited

heads → List<CrossAttention>

final

headSize → int

final

numHeads → int

final

proj → Layer

final

runtimeType → Type

A representation of the runtime type of the object.

no setter, inherited

Methods

forward(List<ValueVector> x_decoder, List<ValueVector> x_encoder) → List<ValueVector>

Forward pass for Multi-Head Cross Attention.

noSuchMethod(Invocation invocation) → dynamic

Invoked when a nonexistent method or property is accessed.

inherited

parameters() → List<Value>

override

toString() → String

A string representation of this object.

inherited

zeroGrad() → void

inherited

Operators

operator ==(Object other) → bool

The equality operator.

inherited