Spatial algorithms and data structures (cupyx.scipy.spatial)#

Note

The spatial module uses pylibraft as a backend. You need to install pylibraft package <https://anaconda.org/rapidsai/pylibraft> from rapidsai Conda channel to use features listed on this page.

Note

Currently, the spatial module is not supported on AMD ROCm platforms.

Nearest-neighbor queries#

KDTree(data[, leafsize, compact_nodes, ...])

KDTree(data, leafsize=16, compact_nodes=True, copy_data=False,

Functions#

distance_matrix(x, y[, p])

Compute the distance matrix.