interpolate

Contents

interpolate#

Interpolation Functions

jrystal.pseudopotential.interpolate.cubic_spline(x: Float[Array, 'x'], y: Float[Array, '*batch_y x'], new_x: Float[Array, '*batch_new_x x']) → Float[Array, '*batch_y_and_batch_new_x x'][source]#

Interpolate a function evaluated at point \(x\) with new inputs using cubic spline. Both \(y\) and new \(x\) can have batch dimensions. Return the interpolated values, which have the same batch dimensions as \(y\) and \(new_x\). The function is not differentiable.

Warning

Cubic spline interpolation is not implemented in JAX. This function uses NumPy and is not differentiable.

Parameters:

  • x (Float[Array, "x"]) – x values.

  • y (Float[Array, “*batch_y x”]) – y values.

  • new_x (Float[Array, “*batch_new_x x”]) – new x values.

Returns:

interpolated y values.

Return type:

Float[Array, “*batch_y *batch_new_x x”]

Contents