scib_rapids.utils.cdist

Contents

scib_rapids.utils.cdist#

scib_rapids.utils.cdist(x, y, metric='euclidean')[source]#

CuPy implementation of pairwise distance computation.

Uses the direct formula for numerical stability (matches JAX/scib-metrics).

Parameters:
  • x (ndarray) – Array of shape (n_cells_a, n_features)

  • y (ndarray) – Array of shape (n_cells_b, n_features)

  • metric (Literal['euclidean', 'cosine'] (default: 'euclidean')) – The distance metric to use: ‘euclidean’ (default) or ‘cosine’.

Return type:

ndarray

Returns:

dist Array of shape (n_cells_a, n_cells_b)