Typo on page 556

[anneleonore]

Hi Sebastian,

Thank you for putting together such an informative book - I find it an enjoyable read and extremely helpful in my research.

Small note: On page 556 you are referring to multihead_keys[2, 1] and say we are accessing the second key value in the second attention head. Should that not be the third attention head?

The accompanying code is: https://github.com/rasbt/machine-learning-book/blob/main/ch16/ch16-part1-self-attention.py

Thanks again!