Masked Softmax Layers in PyTorch · HackerTrans