jax.numpy.interp

jax.numpy.interp(x, xp, fp, left=None, right=None, period=None)

一维线性插值。

JAX 实现 numpy.interp()。

参数:

• x (ArrayLike) – N 维数组，包含要评估插值的 x 坐标。

• xp (ArrayLike) – 一维的已排序数组，包含要进行插值的点。

• fp (ArrayLike) – 形状为 xp.shape 的数组，包含与 xp 关联的函数值。

• left (ArrayLike | str | None) – 指定如何处理 x < xp[0] 的点。默认返回 fp[0]。如果 left 是标量值，则返回该值。如果 left 是字符串 "extrapolate"，则值将通过线性外插确定。left 在指定 period 时将被忽略。

• right (ArrayLike | str | None) – 指定如何处理 x > xp[-1] 的点。默认返回 fp[-1]。如果 right 是标量值，则返回该值。如果 right 是字符串 "extrapolate"，则值将通过线性外插确定。right 在指定 period 时将被忽略。

• period (ArrayLike | None) – 可选指定 x 坐标的周期，例如用于角度空间插值。

返回:

形状为 x.shape 的数组，包含在 x 值处的插值函数。

返回类型:

Array

示例

>>> xp = jnp.arange(10)
>>> fp = 2 * xp
>>> x = jnp.array([0.5, 2.0, 3.5])
>>> interp(x, xp, fp)
Array([1., 4., 7.], dtype=float32)

除非另有说明，否则外插将是常数。

>>> x = jnp.array([-10., 10.])
>>> interp(x, xp, fp)
Array([ 0., 18.], dtype=float32)

使用 "extrapolate" 模式进行线性外插。

>>> interp(x, xp, fp, left='extrapolate', right='extrapolate')
Array([-20.,  20.], dtype=float32)

对于周期性插值，请指定 period。

>>> xp = jnp.array([0, jnp.pi / 2, jnp.pi, 3 * jnp.pi / 2])
>>> fp = jnp.sin(xp)
>>> x = 2 * jnp.pi  # note: not in input array
>>> jnp.interp(x, xp, fp, period=2 * jnp.pi)
Array(0., dtype=float32)