Input
the
monarch
ruled
Embedding Space
king
queen
crown
apple
orange
fruit
Next-Token Probability
queen
0.84
ruler
0.11
throne
0.03
apple
0.01
$\vec{v}_{in}$
$\mathcal{S} \in \mathbb{R}^d$
$P(w_t \mid c_1, \dots, c_n)$