RealIfClose

If input is complex with all imaginary parts close to zero, return real parts.

Abstract Signature:

RealIfClose(a: Array, tol: float = 100)

JAX (Core)

API: jax.numpy.real_if_close
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.real_if_close
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.real_if_close
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.numpy.real_if_close
Strategy: Direct Mapping
Official Docs ↗