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, ...])