Skip to main content

Module vector_ops

Module vector_ops 

Source
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