cr.sparse.pairwise_linf_distances_rw

cr.sparse.pairwise_linf_distances_rw(A, B)

Computes the pairwise Chebyshev 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 Chebyshev distances between points in A and points in B

Return type

(jax.numpy.ndarray)