理解神经网络的注意力机制（Attention）及PyTorch 实现