Array API Standard Support: spatial.distance#

Support on CPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

torch

jax

dask

braycurtis

canberra

cdist

chebyshev

cityblock

correlation

✔️

✔️

✔️

cosine

✔️

✔️

✔️

dice

directed_hausdorff

euclidean

✔️

✔️

✔️

hamming

is_valid_dm

is_valid_y

✔️

✔️

✔️

jaccard

✔️

✔️

✔️

jensenshannon

mahalanobis

minkowski

✔️

✔️

✔️

num_obs_dm

num_obs_y

✔️

✔️

✔️

pdist

rogerstanimoto

russellrao

seuclidean

✔️

✔️

✔️

sokalsneath

sqeuclidean

✔️

✔️

✔️

squareform

yule

Support on GPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

cupy

torch

jax

braycurtis

canberra

cdist

chebyshev

cityblock

correlation

✔️

✔️

✔️

cosine

✔️

✔️

✔️

dice

directed_hausdorff

euclidean

✔️

✔️

✔️

hamming

is_valid_dm

is_valid_y

✔️

✔️

✔️

jaccard

✔️

✔️

✔️

jensenshannon

mahalanobis

minkowski

✔️

✔️

✔️

num_obs_dm

num_obs_y

✔️

✔️

✔️

pdist

rogerstanimoto

russellrao

seuclidean

✔️

✔️

✔️

sokalsneath

sqeuclidean

✔️

✔️

✔️

squareform

yule

Support with JIT#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented