Avoid multiple sqrt computation for text encoder
This commit is contained in:
parent
ff0eaefd9b
commit
8186b38bb6
|
@ -51,8 +51,9 @@ class OnnxTextEncoder {
|
|||
textNormalization += embedding[i] * embedding[i];
|
||||
}
|
||||
|
||||
final double sqrtTextNormalization = sqrt(textNormalization);
|
||||
for (int i = 0; i < 512; i++) {
|
||||
embedding[i] = embedding[i] / sqrt(textNormalization);
|
||||
embedding[i] = embedding[i] / sqrtTextNormalization;
|
||||
}
|
||||
|
||||
return (embedding);
|
||||
|
|
Loading…
Reference in a new issue