pytorch
a4d10342 - [maskedtensor] use masked_softmax for forward/backward instead of regular softmax (#85845)

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
Committer
Parents
Loading