2D · H×W
3D · D×H×W
—
attention scores · per head · per layer
resolution
64²
— memory (fp16)
12 GB GPU →|
—