CdistΒΆ

Computes batched the p-norm distance between each pair of the two collections of row vectors.

Abstract Signature:

Cdist(x1: Tensor, x2: Tensor, p: float = 2.0)

PyTorch

API: torch.cdist
Strategy: Direct Mapping

JAX (Core)

API: jax.scipy.spatial.distance.cdist
Strategy: Direct Mapping