一条推文讲透注意力机制:本质就是矩阵乘法

cneuralnetwork · x · 2026-11-09

cneuralnetwork 用一个最小数值例子拆解多头自注意力:4 个 token、嵌入维度 8、2 个注意力头。输入 X (4×8) 分别乘 Wq/Wk/Wv 得到 Q/K/V,按头拆成 (2,4,4),计算 QKᵀ 后除以 √4 做 softmax,再乘 V,拼接回 (4×8) 并做输出投影,即完成一层多头自注意力(不含残差与归一化)。结论:attention 基本上就是多几步的矩阵乘法。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →