cr.sparse.pdist_l2_rw

cr.sparse.pdist_l2_rw(A)[source]

Computes the pairwise distances between points in A where ach point is a row vector

Parameters

A (jax.numpy.ndarray) – A set of N K-dimensional points (row-wise)

Returns

An NxN matrix D of euclidean distances between points in A

Return type

(jax.numpy.ndarray)