promptbind.models.egnn.unsorted_segment_mean¶ promptbind.models.egnn.unsorted_segment_mean(data, segment_ids, num_segments)[source]¶ Parameters: data – [n_edge, *dimensions] segment_ids – [n_edge] num_segments – [bs * n_node]