cr.sparse.pairwise_l2_distances_rw¶
- cr.sparse.pairwise_l2_distances_rw(A, B)[source]¶
Computes the pairwise distances between points in A and points in B where each point is a row vector
- Parameters
A (jax.numpy.ndarray) – A set of M K-dimensional points (row-wise)
B (jax.numpy.ndarray) – A set of N K-dimensional points (row-wise)
- Returns
An MxN matrix D of euclidean distances between points in A and points in B
- Return type