interpax_fft.interp_rfft2

interpax_fft.interp_rfft2(x0, x1, f, domain0=(0, 6.283185307179586), domain1=(0, 6.283185307179586), axes=(-2, -1))Source

Interpolate real-valued f to coordinates (x0,x1) with FFT and MMT.

Parameters:

  • x0 (jnp.ndarray) – Real query points of coordinate in domain0 where interpolation is desired. Shape must broadcast with shape np.delete(a.shape,axes). The coordinates stored here must be the same coordinate enumerated across axis min(axes) of the function values f.

  • x1 (jnp.ndarray) – Real query points of coordinate in domain1 where interpolation is desired. Shape must broadcast with shape np.delete(a.shape,axes). The coordinates stored here must be the same coordinate enumerated across axis max(axes) of the function values f.

  • f (jnp.ndarray) – Shape (…, f.shape[-2], f.shape[-1]). Real function values on uniform tensor-product grid over an open period.

  • domain0 (tuple[float]) – Domain of coordinate specified by x₀ over which samples were taken.

  • domain1 (tuple[float]) – Domain of coordinate specified by x₁ over which samples were taken.

  • axes (tuple[int]) – Axes along which to transform. The real transform is done along axes[1], so it will be more efficient for that to denote the smaller size axis in axes.

Returns:

fq (jnp.ndarray) – Real function value at query points.