Array API Standard Support: interpolate#

Support on CPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

torch

jax

dask

AAA

Akima1DInterpolator

✔️

✖

✖

BPoly

✔️

✔️

✖

BSpline

✔️

✔️

✖

BarycentricInterpolator

BivariateSpline

N/A

N/A

N/A

CloughTocher2DInterpolator

CubicHermiteSpline

✔️

✔️

✖

CubicSpline

✔️

✔️

✖

FloaterHormannInterpolator

InterpolatedUnivariateSpline

N/A

N/A

N/A

KroghInterpolator

LSQBivariateSpline

N/A

N/A

N/A

LSQSphereBivariateSpline

N/A

N/A

N/A

LSQUnivariateSpline

N/A

N/A

N/A

LinearNDInterpolator

NdBSpline

✔️

✔️

✖

NdPPoly

NearestNDInterpolator

PPoly

✔️

✔️

✖

PchipInterpolator

✖

✖

✖

RBFInterpolator

✔️

✔️

✖

Rbf

N/A

N/A

N/A

RectBivariateSpline

N/A

N/A

N/A

RectSphereBivariateSpline

N/A

N/A

N/A

RegularGridInterpolator

✔️

✔️

✖

SmoothBivariateSpline

N/A

N/A

N/A

SmoothSphereBivariateSpline

N/A

N/A

N/A

UnivariateSpline

N/A

N/A

N/A

approximate_taylor_polynomial

barycentric_interpolate

bisplev

N/A

N/A

N/A

bisplrep

N/A

N/A

N/A

generate_knots

✔️

✔️

✔️

griddata

insert

N/A

N/A

N/A

interp1d

N/A

N/A

N/A

interp2d

interpn

krogh_interpolate

lagrange

make_interp_spline

✔️

✔️

✔️

make_lsq_spline

✔️

✔️

✔️

make_smoothing_spline

✔️

✔️

✔️

make_splprep

✔️

✔️

✔️

make_splrep

✔️

✔️

✔️

pade

pchip_interpolate

spalde

N/A

N/A

N/A

splantider

N/A

N/A

N/A

splder

N/A

N/A

N/A

splev

N/A

N/A

N/A

splint

N/A

N/A

N/A

splprep

N/A

N/A

N/A

splrep

N/A

N/A

N/A

sproot

N/A

N/A

N/A

Support on GPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

cupy

torch

jax

AAA

Akima1DInterpolator

✖

✖

✖

BPoly

✖

✖

✖

BSpline

✖

✖

✖

BarycentricInterpolator

BivariateSpline

N/A

N/A

N/A

CloughTocher2DInterpolator

CubicHermiteSpline

✖

✖

✖

CubicSpline

✖

✖

✖

FloaterHormannInterpolator

InterpolatedUnivariateSpline

N/A

N/A

N/A

KroghInterpolator

LSQBivariateSpline

N/A

N/A

N/A

LSQSphereBivariateSpline

N/A

N/A

N/A

LSQUnivariateSpline

N/A

N/A

N/A

LinearNDInterpolator

NdBSpline

✖

✖

✖

NdPPoly

NearestNDInterpolator

PPoly

✖

✖

✖

PchipInterpolator

✖

✖

✖

RBFInterpolator

✔️

✔️

✔️

Rbf

N/A

N/A

N/A

RectBivariateSpline

N/A

N/A

N/A

RectSphereBivariateSpline

N/A

N/A

N/A

RegularGridInterpolator

✖

✖

✖

SmoothBivariateSpline

N/A

N/A

N/A

SmoothSphereBivariateSpline

N/A

N/A

N/A

UnivariateSpline

N/A

N/A

N/A

approximate_taylor_polynomial

barycentric_interpolate

bisplev

N/A

N/A

N/A

bisplrep

N/A

N/A

N/A

generate_knots

✖

✖

✖

griddata

insert

N/A

N/A

N/A

interp1d

N/A

N/A

N/A

interp2d

interpn

krogh_interpolate

lagrange

make_interp_spline

✖

✖

✖

make_lsq_spline

✖

✖

✖

make_smoothing_spline

✖

✖

✖

make_splprep

✖

✖

✖

make_splrep

✖

✖

✖

pade

pchip_interpolate

spalde

N/A

N/A

N/A

splantider

N/A

N/A

N/A

splder

N/A

N/A

N/A

splev

N/A

N/A

N/A

splint

N/A

N/A

N/A

splprep

N/A

N/A

N/A

splrep

N/A

N/A

N/A

sproot

N/A

N/A

N/A

Support with JIT#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

jax

AAA

Akima1DInterpolator

✖

BPoly

✖

BSpline

✖

BarycentricInterpolator

BivariateSpline

N/A

CloughTocher2DInterpolator

CubicHermiteSpline

✖

CubicSpline

✖

FloaterHormannInterpolator

InterpolatedUnivariateSpline

N/A

KroghInterpolator

LSQBivariateSpline

N/A

LSQSphereBivariateSpline

N/A

LSQUnivariateSpline

N/A

LinearNDInterpolator

NdBSpline

✖

NdPPoly

NearestNDInterpolator

PPoly

✖

PchipInterpolator

✖

RBFInterpolator

✔️

Rbf

N/A

RectBivariateSpline

N/A

RectSphereBivariateSpline

N/A

RegularGridInterpolator

✖

SmoothBivariateSpline

N/A

SmoothSphereBivariateSpline

N/A

UnivariateSpline

N/A

approximate_taylor_polynomial

barycentric_interpolate

bisplev

N/A

bisplrep

N/A

generate_knots

✖

griddata

insert

N/A

interp1d

N/A

interp2d

interpn

krogh_interpolate

lagrange

make_interp_spline

✖

make_lsq_spline

✖

make_smoothing_spline

✖

make_splprep

✖

make_splrep

✖

pade

pchip_interpolate

spalde

N/A

splantider

N/A

splder

N/A

splev

N/A

splint

N/A

splprep

N/A

splrep

N/A

sproot

N/A