Go
pytorch
f0b592da
- Make masked_fill reference traceable (#90972)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
1 year ago
Make masked_fill reference traceable (#90972) As the comment states, `item()` cannot be used since you can't trace through a scalar. Pull Request resolved: https://github.com/pytorch/pytorch/pull/90972 Approved by: https://github.com/ngimel
Author
peterbell10
Committer
pytorchmergebot
Parents
368c7376
Loading