Latent decoder › 6 decoder blocks
Decoder state after attention
×tensor
The residual features after the attention update, feeding the MLP branch and its bypass.
- shape
- [B batch, T positions, 768 d]
Sources
Contextual evidence from Add attention update
A decoder block: Perceiver attention with the queries’ stream first read from the decoder’s state and keys and values from the latents and the state, a scaled residual, an MLP and a second scaled residual.
The implementation, pinned at revision e18c42acdfb7, not yet public
Contextual evidence from MLP
The Perceiver blocks’ MLP: LayerNorm, an affine layer, exact GELU, dropout, an affine layer.
The implementation, pinned at revision e18c42acdfb7, not yet public