diffalign.models.encoder.egnn.unsorted_segment_sum¶

diffalign.models.encoder.egnn.unsorted_segment_sum(data: Tensor, segment_ids: Tensor, num_segments: int) → Tensor[source]¶: TensorFlow-style unsorted_segment_sum for 2D tensors [E, C].