http://zh-v2.d2l.ai/chapter_natural-language-processing-applications/natural-language-inference-attention.html