Array API Standard Support: cluster.hierarchy#

Support on CPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function

torch

jax

dask

average

✔️

✔️

✔️

centroid

✔️

✔️

✔️

complete

✔️

✔️

✔️

cophenet

✔️

✔️

✔️

correspond

✔️

✔️

✔️

cut_tree

✖

✖

✖

dendrogram

✔️

✔️

✔️

fcluster

✔️

✔️

✔️

fclusterdata

✔️

✔️

✔️

from_mlab_linkage

✔️

✔️

✔️

inconsistent

✔️

✔️

✔️

is_isomorphic

✔️

✔️

✔️

is_monotonic

✔️

✔️

✔️

is_valid_im

✔️

✔️

✔️

is_valid_linkage

✔️

✔️

✔️

leaders

✔️

✔️

✔️

leaves_list

✔️

✔️

✔️

linkage

✔️

✔️

✔️

maxRstat

✔️

✔️

✔️

maxdists

✔️

✔️

✔️

maxinconsts

✔️

✔️

✔️

median

✔️

✔️

✔️

num_obs_linkage

✔️

✔️

✔️

optimal_leaf_ordering

✔️

✔️

✔️

set_link_color_palette

N/A

N/A

N/A

single

✔️

✔️

✔️

to_mlab_linkage

✔️

✔️

✔️

to_tree

✔️

✔️

✔️

ward

✔️

✔️

✔️

weighted

✔️

✔️

✔️

Support on GPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function

cupy

torch

jax

average

✖

✖

✖

centroid

✖

✖

✖

complete

✖

✖

✖

cophenet

✖

✖

✖

correspond

✔️

✔️

✔️

cut_tree

✖

✖

✖

dendrogram

✖

✖

✖

fcluster

✖

✖

✖

fclusterdata

✖

✖

✖

from_mlab_linkage

✖

✖

✖

inconsistent

✖

✖

✖

is_isomorphic

✔️

✔️

✔️

is_monotonic

✔️

✔️

✔️

is_valid_im

✔️

✔️

✔️

is_valid_linkage

✔️

✔️

✔️

leaders

✖

✖

✖

leaves_list

✖

✖

✖

linkage

✖

✖

✖

maxRstat

✖

✖

✖

maxdists

✖

✖

✖

maxinconsts

✖

✖

✖

median

✖

✖

✖

num_obs_linkage

✔️

✔️

✔️

optimal_leaf_ordering

✖

✖

✖

set_link_color_palette

N/A

N/A

N/A

single

✖

✖

✖

to_mlab_linkage

✔️

✔️

✔️

to_tree

✔️

✔️

✔️

ward

✖

✖

✖

weighted

✖

✖

✖

Support with JIT#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function

jax

average

✔️

centroid

✔️

complete

✔️

cophenet

✔️

correspond

✔️

cut_tree

✖

dendrogram

✖

fcluster

✖

fclusterdata

✖

from_mlab_linkage

✔️

inconsistent

✔️

is_isomorphic

✔️

is_monotonic

✔️

is_valid_im

✔️

is_valid_linkage

✔️

leaders

✖

leaves_list

✔️

linkage

✔️

maxRstat

✔️

maxdists

✔️

maxinconsts

✔️

median

✔️

num_obs_linkage

✔️

optimal_leaf_ordering

✔️

set_link_color_palette

N/A

single

✔️

to_mlab_linkage

✔️

to_tree

✖

ward

✔️

weighted

✔️