mx.symbol.argsort

Description

Returns the indices that would sort an input array along the given axis.

This function performs sorting along the given axis and returns an array of indices having same shape as an input array that index data in sorted order.

Example:

x = [[ 0.3,  0.2,  0.4],
[ 0.1,  0.3,  0.2]]

// sort along axis -1
argsort(x) = [[ 1.,  0.,  2.],
[ 0.,  2.,  1.]]

// sort along axis 0
argsort(x, axis=0) = [[ 1.,  0.,  1.]
[ 0.,  1.,  0.]]

// flatten and then sort
argsort(x, axis=None) = [ 3.,  1.,  5.,  0.,  4.,  2.]

Usage

mx.symbol.argsort(...)

Arguments

Argument

Description

data

NDArray-or-Symbol.

The input array

axis

int or None, optional, default=’-1’.

Axis along which to sort the input tensor. If not given, the flattened array is used. Default is -1.

is.ascend

boolean, optional, default=1.

Whether to sort in ascending or descending order.

dtype

{‘float16’, ‘float32’, ‘float64’, ‘int32’, ‘int64’, ‘uint8’},optional, default=’float32’.

DType of the output indices. It is only valid when ret_typ is “indices” or “both”. An error will be raised if the selected data type cannot precisely represent the indices.

name

string, optional.

Name of the resulting symbol.