jax.numpy.sort_complex

jax.numpy.sort_complex(a)

返回复数数组的排序副本。

JAX 实现的 numpy.sort_complex()。

复数按照字典顺序排序，即首先按实部排序，如果实部相等，则按虚部排序。

参数：

a (ArrayLike) – 输入数组。如果 dtype 不是复数，则数组将向上转换为复数。

返回：

与输入具有相同形状和复数 dtype 的排序数组。如果 a 是多维的，则沿最后一个轴排序。

返回类型：

Array

另请参阅

• 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)