Skip to content

Is there any method to implement scatter_quantile #502

@DulyHao

Description

@DulyHao

Hi there,
I'm currently working on a PyTorch project that requires group-wise quantile calculation (e.g., median, quartiles) based on grouping indices, which is similar to the core functionality of torch_scatter (like scatter_mean, scatter_sum), but for quantile statistics.
I notice that torch_scatter provides rich scatter aggregation methods for common statistics (sum/mean/max/min etc.), but there is no official scatter_quantile API yet. I'm wondering:
Is there any existing way (or workaround) to implement scatter_quantile efficiently with the current torch_scatter APIs (e.g., scatter_sort for grouped sorting)?
Does the development team have plans to add scatter_quantile as an official API in future versions of torch_scatter?

Kind regards

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions