KV Cache 三维张量 — 一个 Token 的 K(或 V)长什么样?
一个 Token 的 K(或 V)长什么样?— [L × H_kv × D] 三维张量
L = 28 层
head 0
head 1
head 2
head 3
H_kv=4
D = 128 (简化显示 16 格)
0.3
每个格子 = 1 个 FP16 值 = 2 bytes
汇总: 一个 Token 的 K(或 V)占多少显存?
前面矩阵 (1 层): H_kv × D × B
= 4 × 128 × 2 = 1,024 Bytes
L 层 (28 层): L × H_kv × D × B
= 28 × 4 × 128 × 2 = 28,672 Bytes ≈ 28 KiB
K+V 两份: 2 × L × H_kv × D × B
= 57,344 Bytes ≈ 56 KiB
T=2048 tokens: T × 56 KiB
≈ 112 MiB