Skip to content

Commit 15b90b7

Browse files
pulsejetxenova
andauthored
maths: switch dot to faster for loop (#667)
* maths: switch dot to faster for loop * Update src/utils/maths.js --------- Co-authored-by: Joshua Lochner <[email protected]>
1 parent cdbc532 commit 15b90b7

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/utils/maths.js

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,11 @@ export function log_softmax(arr) {
174174
* @returns {number} The dot product of arr1 and arr2.
175175
*/
176176
export function dot(arr1, arr2) {
177-
return arr1.reduce((acc, val, i) => acc + val * arr2[i], 0);
177+
let result = 0;
178+
for (let i = 0; i < arr1.length; ++i) {
179+
result += arr1[i] * arr2[i];
180+
}
181+
return result;
178182
}
179183

180184

0 commit comments

Comments
 (0)