JaxBackend.argsort

JaxBackend.argsort(*args, **kwargs)

Compute the indices to sort a given input array.

Parameters:

arrBackendArray

Input array.

dtypetype

Target data type.

Returns:

BackendArray

Indices that would sort the input data.