pytorch 实现 transformer

attention 是一种我们常用到的注意力机制。