pytorch
ed46b967 - add randomness kwarg to jacfwd (#84220)

Commit
3 years ago
add randomness kwarg to jacfwd (#84220) From https://github.com/pytorch/functorch/issues/1010, if a user runs jacfwd with a function that uses randomness, it will fail since the default behavior for vmap is error. This lets the user specify the randomness behavior to jacfwd too since it is doing vmap(jvp(forward)). This is less likely to show up in jacrev since that only vmaps over the backwards pass Pull Request resolved: https://github.com/pytorch/pytorch/pull/84220 Approved by: https://github.com/zou3519
Author
samdow
Committer
Parents
Loading