jax.numpy.modf

jax.numpy.modf(x, /, out=None)

返回输入数组的逐元素小数部分和整数部分。

numpy.modf 的 JAX 实现。

参数:

• x (ArrayLike) – 输入数组或标量。

• out – JAX 不使用。

返回:

一个数组，包含 x 元素的小数部分和整数部分，并提升不精确的 dtype。

返回类型:

tuple[Array, Array]

另请参阅

• jax.numpy.divmod()：逐元素计算 x1 除以 x2 的整数商和余数。

示例

>>> jnp.modf(4.8)
(Array(0.8000002, dtype=float32, weak_type=True), Array(4., dtype=float32, weak_type=True))
>>> x = jnp.array([-3.4, -5.7, 0.6, 1.5, 2.3])
>>> jnp.modf(x)
(Array([-0.4000001 , -0.6999998 ,  0.6       ,  0.5       ,  0.29999995],      dtype=float32), Array([-3., -5.,  0.,  1.,  2.], dtype=float32))