pytorch
a4d10342
- [maskedtensor] use masked_softmax for forward/backward instead of regular softmax (#85845)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[maskedtensor] use masked_softmax for forward/backward instead of regular softmax (#85845) Pull Request resolved: https://github.com/pytorch/pytorch/pull/85845 Approved by: https://github.com/cpuhrsch
Author
george-qi
Committer
pytorchmergebot
Parents
1c970846
Loading