Add dot function for good measure

tekknolagi · Sep 23, 2023 · 1daf1aa · 1daf1aa
1 parent 416040e
commit 1daf1aa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/_posts/2023-09-19-compiling-ml-models.md b/_posts/2023-09-19-compiling-ml-models.md
@@ -1276,6 +1276,9 @@ nice speedup.
 The code for compiling a `Dot` node is not that tricky:
 
 ```python
+def dot(l, r):
+    return sum(li.data*ri.data for li,ri in zip(l,r))
+
 class Dot(Value):
     def __init__(self, left_arr, right_arr):
         assert len(left_arr) == len(right_arr)