jax.numpy.sort_complex

jax.numpy.sort_complex(a)

傳回複數陣列的排序副本。

JAX 版本的 numpy.sort_complex() 實作。

複數依字典順序排序，先依實部排序，若實部相等則依虛部排序。

參數:

a (ArrayLike) – 輸入陣列。如果 dtype 不是複數，陣列將會向上轉型為複數。

傳回值:

與輸入形狀相同且 dtype 為複數的排序陣列。如果 a 是多維的，則沿著最後一個軸排序。

傳回型別:

陣列

另請參閱

• jax.numpy.sort()：傳回陣列的排序副本。

範例

>>> a = jnp.array([1+2j, 2+4j, 3-1j, 2+3j])
>>> jnp.sort_complex(a)
Array([1.+2.j, 2.+3.j, 2.+4.j, 3.-1.j], dtype=complex64)

多維陣列沿著最後一個軸排序

>>> a = jnp.array([[5, 3, 4],
...                [6, 9, 2]])
>>> jnp.sort_complex(a)
Array([[3.+0.j, 4.+0.j, 5.+0.j],
       [2.+0.j, 6.+0.j, 9.+0.j]], dtype=complex64)