pytorch

e289a18e - Support multi-dimensional lengths in segment_reduce to support pytorch_scatter.segment_* functionalities (CPU-only)

Support multi-dimensional lengths in segment_reduce to support pytorch_scatter.segment_* functionalities (CPU-only) Pull Request resolved: https://github.com/pytorch/pytorch/pull/77058 Approved by: https://github.com/cpuhrsch