cr.sparse.pairwise_l2_distances_cw

cr.sparse.pairwise_l2_distances_cw(A, B)[source]

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

Parameters
Returns

An MxN matrix D of euclidean distances between points in A and points in B

Return type

(jax.numpy.ndarray)