[system-a] linear

xieyuheng · May 25, 2024 · eca77ab · eca77ab
1 parent c8c4a68
commit eca77ab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/TODO.md b/TODO.md
@@ -1,6 +1,7 @@
 # system-a
 
-[system-a] `linear`
+[system-a] test `linear`
+[system-a] `relu`
 
 # the-book
 

diff --git a/src/system-a/toys/linear.ts b/src/system-a/toys/linear.ts
@@ -0,0 +1,7 @@
+import type { Tensor } from "../tensor/index.js"
+import { dot } from "./dot.js"
+import { add } from "./toys.js"
+
+export function linear(t: Tensor): (weight: Tensor, bias: Tensor) => Tensor {
+  return (weight, bias) => add(dot(weight, t), bias)
+}