pytorch

40f7ef1f - Support multi-dimensional lengths in segment_reduce to support pytorch_scatter.segment_* functionalities (CUDA)

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