Expand description
Vector Operations
Helpers for vector similarity search and distance computation
Functionsยง
- compute_
vector_ distances - Compute vector distance for an array of vectors against a single query vector
- compute_
vector_ similarities - Compute vector similarities for an array of vectors against a single query vector
- cosine_
distance - Compute cosine distance (1 - cosine_similarity) between two vectors Returns a value in [0, 2] where 0 means identical and 2 means opposite
- cosine_
similarity - Compute cosine similarity (for vector_similarity function) Returns a value in [-1, 1] where 1 means identical and -1 means opposite
- dot_
product_ distance - Compute dot product between two vectors For similarity search, we return the negative (so lower is better for sorting)
- dot_
product_ similarity - Compute dot product similarity (for vector_similarity function)
- extract_
single_ vector_ from_ scalar - Extract a single vector from a ScalarValue This avoids allocating a full array when we just need one vector
- extract_
vectors - Extract vectors from Arrow ListArray or FixedSizeListArray
- l2_
distance - Compute L2 (Euclidean) distance between two vectors