Attention Pooling: Nadaraya-Watson Kernel Regression

https://d2l.ai/chapter_attention-mechanisms/nadaraya-watson.html