SortComplexΒΆ

Sort a complex array using the real part first, then the imaginary part.

Abstract Signature:

SortComplex(a: Array)

PyTorch

API: β€”
Strategy: Custom / Partial

JAX (Core)

API: jax.numpy.sort_complex
Strategy: Direct Mapping

NumPy

API: numpy.sort_complex
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.sort_complex
Strategy: Direct Mapping

PaxML / Praxis

API: jax.numpy.sort_complex
Strategy: Direct Mapping