Embeddings

Layer 2 · Mechanics

Embeddings

How embeddings are actually learned via lookup tables and training signal, and how similarity is measured and used.

8 min read70 XP

  1. token id (integer)

  2. lookup row in embedding table

  3. dense vector

  4. used by rest of the network

An embedding layer is literally a big lookup table of learned vectors, one row per vocabulary item.

Mechanically, an embedding layer is just a matrix (vocabulary size , embedding dimension ). Looking up token means reading row of . Critically, 's rows are learned parameters updated by gradient descent, just like any other weight matrix — the network discovers useful vector positions purely by trying to reduce its training loss.

Go deeper: L3 Code