catniff 0.1.1 → 0.1.2

package/README.md

@@ -1,4 +1,4 @@
-## Catniff
+# Catniff
 
 Catniff is a small and experimental tensor library and autograd engine inspired by [micrograd](https://github.com/karpathy/micrograd). The name is a play on "catnip" and "differentiation".
 
@@ -41,13 +41,26 @@ All available APIs are in `./src/tensor.ts`.
 
 To compute the gradient of our mathematical expression, we use the `Node` class to dynamically build our DAG:
 ```js
-const { Node } = require("catniff");
+const { Node } = require("../index");
 
-const X = new Node([ 1, 2, 3 ]);
-const L = x.pow(2).add(x); // X^2 + X
+const X = new Node([
+    [ 0.5, -1.0 ],
+    [ 2.0,  0.0 ]
+]);
 
-L.backward();
-console.log(x.grad);
+const Y = new Node([
+    [ 1.0, -2.0 ],
+    [ 0.5,  1.5 ]
+]);
+
+const D = X.sub(Y);
+const E = D.exp();
+const F = E.add(1);
+const G = F.log();
+
+G.backward();
+
+console.log(X.grad, Y.grad);
 ```
 
 All available APIs are in `./src/autograd.ts`.

package/dist/tensor.d.ts

@@ -11,6 +11,7 @@ export declare class TensorMath {
     static lt(tA: Tensor, tB: Tensor): Tensor;
     static ge(tA: Tensor, tB: Tensor): Tensor;
     static le(tA: Tensor, tB: Tensor): Tensor;
+    static eq(tA: Tensor, tB: Tensor): Tensor;
     static neg(tA: Tensor): Tensor;
     static exp(tA: Tensor): Tensor;
     static log(tA: Tensor): Tensor;

package/dist/tensor.js

@@ -247,6 +247,31 @@ class TensorMath {
         }
         throw new Error("Inputs are not tensors");
     }
+    static eq(tA, tB) {
+        if (typeof tA === "number" && typeof tB === "number") {
+            return tA === tB ? 1 : 0;
+        }
+        else if (Array.isArray(tA) && Array.isArray(tB)) {
+            const outLen = Math.max(tA.length, tB.length);
+            if (tA.length !== tB.length && tA.length !== 1 && tB.length !== 1) {
+                throw new Error("Inputs are incompatible tensors");
+            }
+            const result = [];
+            for (let i = 0; i < outLen; i++) {
+                const subA = tA[tA.length === 1 ? 0 : i];
+                const subB = tB[tB.length === 1 ? 0 : i];
+                result.push(TensorMath.eq(subA, subB));
+            }
+            return result;
+        }
+        else if (Array.isArray(tA) && typeof tB === "number") {
+            return tA.map(subA => TensorMath.eq(subA, tB));
+        }
+        else if (typeof tA === "number" && Array.isArray(tB)) {
+            return tB.map(subB => TensorMath.eq(tA, subB));
+        }
+        throw new Error("Inputs are not tensors");
+    }
     static neg(tA) {
         if (typeof tA === "number") {
             return -tA;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "catniff",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "A cute autograd engine for Javascript",
   "main": "index.js",
   "scripts": {