61 transformer blocks › 58 MoE blocks › Sparse latent attention › Sparse attention › Mix and project
Merge heads
×no parametersx.flatten(2)
Concatenate the heads at each position.
- in
- [B batch, 128 heads, T positions, 128 value]
- out
- [B batch, T positions, 16384 heads × value]
Explanation
Heads are concatenated in order: head fills features .
Codemodel.py 607–608
x = self.wo(x.flatten(2))
return xSources1
Direct evidence
Head outputs are flattened and projected back to the model width.
mla-output · deepseek-ai/DeepSeek-V3.2-Exp / inference/model.py
Pinned revision 87e509a2e5a1 ↗ · lines 607–608
Read code excerpt
x = self.wo(x.flatten(2))
return x