https://d2l.ai/chapter_natural-language-processing-applications/natural-language-inference-attention.html