mlx_graphs.utils.sorting.sort_edge_index_and_features#
- mlx_graphs.utils.sorting.sort_edge_index_and_features(edge_index: mlx.core.array, edge_features: mlx.core.array) tuple[mlx.core.array, mlx.core.array] [source]#
Sorts the given edge_index and their corresponding features.
- Parameters:
- Return type:
- Returns:
A tuple containing the sorted edge index and the corresponding sorted edge features.