jax.lax.sort#

jax.lax.sort(operand: Array, dimension: int = -1, is_stable: bool = True, num_keys: int = 1) → Array[原始碼]#
jax.lax.sort(operand: Sequence[Array], dimension: int = -1, is_stable: bool = True, num_keys: int = 1) → tuple[Array, ...]

包裝 XLA 的 Sort 運算子。

對於浮點輸入，-0.0 和 0.0 被視為相等，而 NaN 值會排序到陣列的末尾。對於複數輸入，排序順序是實部和虛部的詞典編纂順序，實部為主。

參數:

  • operand – 陣列或陣列序列

  • dimension – 沿其排序的整數維度。預設值：-1。

  • is_stable – 布林值，指定是否使用穩定排序。預設值：True。

  • num_keys – 要視為排序鍵的運算元數量。預設值：1。對於 num_keys > 1，排序順序將使用前 num_keys 個陣列以詞典編纂方式確定，第一個鍵為主鍵。其餘運算元將以相同的排列方式傳回。

傳回:

輸入或輸入的排序版本。

傳回類型:

operand