Distance computations (cupyx.scipy.spatial.distance)#


The distance 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.


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

Distance matrix computations#

Distance matrix computation from a collection of raw observation vectors stored in a rectangular array.

cdist(XA, XB[, metric, out])

Compute distance between each pair of the two collections of inputs.

Distance functions#

Distance functions between two numeric vectors u and v. Computing distances over a large collection of vectors is inefficient for these functions. Use cdist for this purpose.

minkowski(u, v, p)

Compute the Minkowski distance between two 1-D arrays.