MultiHeadAttention

Record fields

NameDescription
HeadDim
NumHeads
WK
WO
WQ
WV

Instance members

NameDescription
this.forward (x, ?mask, ?kvCache)

PreNormTransformerBlock

Pre-norm transformer block: Norm → Attn → Add → Norm → FFN → Add.

Record fields

NameDescription
Attn
AttnNorm
Ff1
Ff2
FfNorm

Instance members

NameDescription
this.forward (x, ?mask)

PostNormTransformerBlock

Post-norm transformer block: Attn → Add → Norm → FFN → Add → Norm.

Record fields

NameDescription
Attn
AttnNorm
Ff1
Ff2
FfNorm

Instance members

NameDescription
this.forward (x, ?mask)x←LN(x+Attn(x))x \leftarrow \mathrm{LN}(x + \mathrm{Attn}(x)), then x←LN(x+FFN(x))x \leftarrow \mathrm{LN}(x + \mathrm{FFN}(x)).