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

Removed pow

Browse Source
This commit is contained in:
Andrew Kane
2021-04-24 17:15:53 -07:00
parent fb015adc25
commit 2515cf050d
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/vector.c
View File

@@ -361,11 +361,15 @@ l2_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);
double distance = 0.0; double distance = 0.0;
double diff;
CheckDims(a, b); CheckDims(a, b);
for (int i = 0; i < a->dim; i++) for (int i = 0; i < a->dim; i++)
distance += pow(a->x[i] - b->x[i], 2); {
diff = a->x[i] - b->x[i];
distance += diff * diff;
}
PG_RETURN_FLOAT8(sqrt(distance)); PG_RETURN_FLOAT8(sqrt(distance));
} }
@@ -381,11 +385,15 @@ vector_l2_squared_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);
double distance = 0.0; double distance = 0.0;
double diff;
CheckDims(a, b); CheckDims(a, b);
for (int i = 0; i < a->dim; i++) for (int i = 0; i < a->dim; i++)
distance += pow(a->x[i] - b->x[i], 2); {
diff = a->x[i] - b->x[i];
distance += diff * diff;
}
PG_RETURN_FLOAT8(distance); PG_RETURN_FLOAT8(distance);
} }
@@ -446,8 +454,8 @@ cosine_distance(PG_FUNCTION_ARGS)
for (int i = 0; i < a->dim; i++) for (int i = 0; i < a->dim; i++)
{ {
distance += a->x[i] * b->x[i]; distance += a->x[i] * b->x[i];
norma += pow(a->x[i], 2); norma += a->x[i] * a->x[i];
normb += pow(b->x[i], 2); normb += b->x[i] * b->x[i];
} }
PG_RETURN_FLOAT8(1 - (distance / (sqrt(norma) * sqrt(normb)))); PG_RETURN_FLOAT8(1 - (distance / (sqrt(norma) * sqrt(normb))));
@@ -503,7 +511,7 @@ vector_norm(PG_FUNCTION_ARGS)
double norm = 0.0; double norm = 0.0;
for (int i = 0; i < a->dim; i++) for (int i = 0; i < a->dim; i++)
norm += pow(a->x[i], 2); norm += a->x[i] * a->x[i];
PG_RETURN_FLOAT8(sqrt(norm)); PG_RETURN_FLOAT8(sqrt(norm));
} }