mlx.core.argsort

目录

mlx.core.argsort#

argsort(a: array, /, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#

Returns the indices that sort the array.

参数:

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).

返回:

The uint32 array containing indices that sort the input.

返回类型: