torch.masked_fill() - beyondnlp/nlp GitHub Wiki