mlx.core.argsort# argsort(a: array, /, axis: None | int = -1, *, stream: None | Stream | Device = None) → array# Returns the indices that sort the array. Parameters: a (array) – Input array. axis (int or None, optional) – Optional axis to sort over. If None, this sorts over the flattened array. If unspecified, it defaults to -1 (sorting over the last axis). Returns: The uint32 array containing indices that sort the input. Return type: array
