Switch language한국어
Back to the list

Understanding Transformers Part 17: Generating the Output Word

TL;DR AI

Key summary

1 min read
  1. A Transformer decoder turns its final hidden states into vocabulary scores with a dense layer.

  2. Softmax selects the most likely output word from the vocabulary.

  3. The predicted word is fed back into the decoder to generate the next token.

  4. This process repeats until an end-of-sequence token is produced, completing the sentence.

Read the original