cr.sparse.pdist_l2_cw

cr.sparse.pdist_l2_cw(A)[source]

Computes the pairwise distances between points in A where each point is a column vector

Parameters

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

Returns

An NxN matrix D of euclidean distances between points in A

Return type

(jax.numpy.ndarray)