jax.numpy.frexp

jax.numpy.frexp(x, /)

将浮点数值拆分为尾数和以 2 为底的指数。

numpy.frexp() 的 JAX 实现。

参数:

x (ArrayLike) – 实值数组

返回:

一个元组 (mantissa, exponent)，其中 mantissa 是 -1 和 1 之间的浮点值，exponent 是一个整数，使得 x == mantissa * 2 ** exponent。

返回类型:

元组[Array, Array]

另请参阅

• jax.numpy.ldexp()：计算 frexp 的逆运算。

示例

将值拆分为尾数和指数

>>> x = jnp.array([1., 2., 3., 4., 5.])
>>> m, e = jnp.frexp(x)
>>> m
Array([0.5  , 0.5  , 0.75 , 0.5  , 0.625], dtype=float32)
>>> e
Array([1, 2, 2, 3, 3], dtype=int32)

重建原始数组

>>> m * 2 ** e
Array([1., 2., 3., 4., 5.], dtype=float32)