jax.numpy.lexsort

jax.numpy.lexsort(keys, axis=-1)

使用一系列键执行间接稳定排序。

LAX 后端实现 numpy.lexsort().

原始文档字符串如下。

给定多个排序键，lexsort 返回一个整数索引数组，该数组描述按多个键的排序顺序。序列中的最后一个键用于主要排序顺序，使用倒数第二个键来打破平局，依此类推。

参数：

• keys ((k, m, n, ...) array-like) – 要排序的 k 个键。序列中的最后一个键（例如，如果 keys 是一个二维数组，则为最后一行）是主要排序键。 keys 沿零轴的每个元素必须是相同形状的类数组对象。

• axis (int, optional) – 要间接排序的轴。默认情况下，对每个序列的最后一个轴进行排序。沿 axis 的独立切片分别排序；请参见最后一个示例。

返回：

indices – 按指定轴排序键的索引数组。

返回类型：

(m, n, …) 整数 ndarray