pytorch

d046376c - Dispatch `numpy.take_along_axis` to `torch.take_along_dim` (#108880)

Dispatch `numpy.take_along_axis` to `torch.take_along_dim` (#108880) Pull Request resolved: https://github.com/pytorch/pytorch/pull/108880 Approved by: https://github.com/lezcano ghstack dependencies: #108879