pytorch

f969834f - [functorch] vmap: nansum & nanmean (#91372)

[functorch] vmap: nansum & nanmean (#91372) Fixes #91174 Pull Request resolved: https://github.com/pytorch/pytorch/pull/91372 Approved by: https://github.com/zou3519