smiet.jax.utilities.batched_interpolation module

smiet.jax.utilities.batched_interpolation.batched_interp1d(x, xp, fp)

Batched linear interpolation with boundary handling.

For left and right values, the values at the boundaries are used.

Parameters:
x : jnp.ndarray

Input points to interpolate. Shape: (Nant, Nslice).

xp : jnp.ndarray

Points to interpolate at. Shape: (Ngrid,).

fp : jnp.ndarray

Function values at xp. Shape: (Ngrid, Nfreq, Nslice).

Returns:

Interpolated values. Shape: (Nant, Nfreq, Nslice).

Return type:

jnp.ndarray

smiet.jax.utilities.batched_interpolation.cubic_interp1d(x, xp, fp)
smiet.jax.utilities.batched_interpolation.linear_interp_with_bounds(x, xp, fp)