liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-11 07:06:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Moved vector cosine similarity to separate function [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-15 10:33:23 -07:00
parent ba99255bbb
commit fac88d0fd4
1 changed files with 20 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/vector.c
View File

@@ -653,6 +653,25 @@ vector_negative_inner_product(PG_FUNCTION_ARGS)
PG_RETURN_FLOAT8((double) -VectorInnerProduct(a->dim, a->x, b->x)); PG_RETURN_FLOAT8((double) -VectorInnerProduct(a->dim, a->x, b->x));
} }
static double
VectorCosineSimilarity(int dim, float *ax, float *bx)
{
float similarity = 0.0;
float norma = 0.0;
float normb = 0.0;
/* Auto-vectorized */
for (int i = 0; i < dim; i++)
{
similarity += ax[i] * bx[i];
norma += ax[i] * ax[i];
normb += bx[i] * bx[i];
}
/* Use sqrt(a * b) over sqrt(a) * sqrt(b) */
return (double) similarity / sqrt((double) norma * (double) normb);
}
/* /*
* Get the cosine distance between two vectors * Get the cosine distance between two vectors
*/ */
@@ -662,25 +681,11 @@ cosine_distance(PG_FUNCTION_ARGS)
{ {
Vector *a = PG_GETARG_VECTOR_P(0); Vector *a = PG_GETARG_VECTOR_P(0);
Vector *b = PG_GETARG_VECTOR_P(1); Vector *b = PG_GETARG_VECTOR_P(1);
float *ax = a->x;
float *bx = b->x;
float distance = 0.0;
float norma = 0.0;
float normb = 0.0;
double similarity; double similarity;
CheckDims(a, b); CheckDims(a, b);
/* Auto-vectorized */ similarity = VectorCosineSimilarity(a->dim, a->x, b->x);
for (int i = 0; i < a->dim; i++)
{
distance += ax[i] * bx[i];
norma += ax[i] * ax[i];
normb += bx[i] * bx[i];
}
/* Use sqrt(a * b) over sqrt(a) * sqrt(b) */
similarity = (double) distance / sqrt((double) norma * (double) normb);
#ifdef _MSC_VER #ifdef _MSC_VER
/* /fp:fast may not propagate NaN */ /* /fp:fast may not propagate NaN */