Transformer 源码中 Mask 机制的实现