pytorch

371d8ba5 - vmap: decompose real and imag instead of registering batch rule (#110508)

vmap: decompose real and imag instead of registering batch rule (#110508) Clean-up Pull Request resolved: https://github.com/pytorch/pytorch/pull/110508 Approved by: https://github.com/zou3519