jax.scipy.interpolate.RegularGridInterpolator

jax.scipy.interpolate.RegularGridInterpolator#

class jax.scipy.interpolate.RegularGridInterpolator(points, values, method='linear', bounds_error=False, fill_value=nan)[源代码][源代码]#

在规则的矩形网格上插值点。

JAX 实现的 scipy.interpolate.RegularGridInterpolator()

参数:
  • points – 指定网格坐标的N个数组序列。

  • values – 指定网格值的 N 维数组。

  • method – 插值方法,可以是 "linear""nearest"

  • bounds_error – JAX 未实现

  • fill_value – 网格外的点的返回值,默认为 NaN。

返回:

可调用的插值对象。

返回类型:

interpolator

示例

>>> points = (jnp.array([1, 2, 3]), jnp.array([4, 5, 6]))
>>> values = jnp.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]])
>>> interpolate = RegularGridInterpolator(points, values, method='linear')
>>> query_points = jnp.array([[1.5, 4.5], [2.2, 5.8]])
>>> interpolate(query_points)
Array([30., 64.], dtype=float32)
__init__(points, values, method='linear', bounds_error=False, fill_value=nan)[源代码][源代码]#

方法

__init__(points, values[, method, ...])