nn.Transformer explaination

[zhangguanheng66 (Guanheng George Zhang)]

For your first part, it seems that you are not setting up attn_mask correctly.