pytorch

a5d9d2aa - [functorch] remove argnums partial helper function, rewrite test to use slice argnum (#84951)

[functorch] remove argnums partial helper function, rewrite test to use slice argnum (#84951) Pull Request resolved: https://github.com/pytorch/pytorch/pull/84951 Approved by: https://github.com/zou3519