Pytorch transformer forward function masks implementation for decoder forward function

Question

I am trying to use and learn PyTorch Transformer with DeepMind math dataset. I have tokenized (char not word) sequence that is fed into model. Models forward function is doi