enerzyme.models.functional.segment_sum_coo

enerzyme.models.functional.segment_sum_coo(src: Tensor, idx: Tensor, dim_size: int | None = None) → Tensor