interpax_fft.irfft2_mmt_pos

interpax_fft.irfft2_mmt_pos(x0, x1, a, n0, n1, domain0=(0, 6.283185307179586), domain1=(0, 6.283185307179586), axes=(-2, -1))Source

Evaluate Fourier coefficients a at coordinates (x0,x1).

Uses matrix multiplication transform.

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 Fourier coefficients a.

  • 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 Fourier coefficients a.

  • a (jnp.ndarray) – Shape (…, a.shape[-2], a.shape[-1]). Fourier coefficients. f=rfft2(f,axes=axes,norm="forward") a=jnp.moveaxis(f,axes,(-2,-1)).at[...,i].divide(2)*2.

  • n0 (int) – Spectral resolution of a for domain0.

  • n1 (int) – Spectral resolution of a for domain1.

  • 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.

Returns:

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