[PyTorch] Add & use inferExpandGeometry_dimvector (#55316)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55316
No need for heap allocations in the common case here.
ghstack-source-id: 126170054
Test Plan: Existing CI
Reviewed By: hlu1
Differential Revision: D27571942
fbshipit-source-id: 11fbf077c583c80ea63e024d2b9e1599785fff71
