jaxrts.weissker_interpolator.VRegularGridLinearInterpolator

class jaxrts.weissker_interpolator.VRegularGridLinearInterpolator(points, values)[source]

Vectorized RegularGridInterpolator, with a linear interpolation: Interpolated linarly over the last indices of values, i.e., values.shape should be [*vectorized_dims.shape, *points.shape], and the output of a call will be of shape vectorized_dims.shape.

Methods

__init__(points, values)

tree_flatten()

tree_unflatten(aux_data, children)