pytorch

a67fa845 - [vmap] Fix searchsorted batch rule (#99698)

[vmap] Fix searchsorted batch rule (#99698) Hopefully there is no other missed cases. Fixes #99603 Pull Request resolved: https://github.com/pytorch/pytorch/pull/99698 Approved by: https://github.com/kshitij12345, https://github.com/zou3519