[PyTorch] Flip polarity of masked_softmax mask (#78)
Summary:
X-link: https://github.com/pytorch/pytorch-canary/pull/78
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75039
It didn't match torch.nn.MultiheadAttention. Now it does.
ghstack-source-id: 152815449
Test Plan: updated tests
Reviewed By: zrphercule
Differential Revision: D34929186
fbshipit-source-id: 1eaee615bafd5a6f058f1faefa54f8f4aa01c92e
(cherry picked from commit 00eea72a06fb924112f1036c8f0c5ed08eb0d02c)
