jax.scipy.interpolate.RegularGridInterpolator

jax.scipy.interpolate.RegularGridInterpolator#

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

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

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)[source]#

方法

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

内容