catniff 0.1.7 → 0.1.8

package/dist/autograd.d.ts

@@ -45,7 +45,8 @@ export declare enum OP {
     SIGMOID = 42,
     TANH = 43,
     T = 44,
-    MM = 45
+    MM = 45,
+    DOT = 46
 }
 export declare class Node {
     value: Tensor;
@@ -100,6 +101,7 @@ export declare class Node {
     tanh(): Node;
     t(): Node;
     mm(other: Node | number): Node;
+    dot(other: Node | number): Node;
     backward(): void;
     static forceNode(value: Node | number): Node;
     static addGrad(node: Node, accumGrad: Tensor): void;

package/dist/autograd.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Node = exports.OP = void 0;
 const tensor_1 = require("./tensor");
-const { add, sub, mul, pow, div, gt, lt, ge, le, eq, logicalAnd, logicalOr, logicalXor, logicalNot, bitwiseAnd, bitwiseOr, bitwiseXor, bitwiseNot, bitwiseLeftShift, bitwiseRightShift, neg, abs, sign, sin, cos, tan, asin, acos, atan, sinh, cosh, asinh, acosh, atanh, sqrt, exp, log, log2, log10, log1p, relu, sigmoid, tanh, t, mm } = tensor_1.TensorMath;
+const { add, sub, mul, pow, div, gt, lt, ge, le, eq, logicalAnd, logicalOr, logicalXor, logicalNot, bitwiseAnd, bitwiseOr, bitwiseXor, bitwiseNot, bitwiseLeftShift, bitwiseRightShift, neg, abs, sign, sin, cos, tan, asin, acos, atan, sinh, cosh, asinh, acosh, atanh, sqrt, exp, log, log2, log10, log1p, relu, sigmoid, tanh, t, mm, dot } = tensor_1.TensorMath;
 var OP;
 (function (OP) {
     OP[OP["NONE"] = 0] = "NONE";
@@ -51,6 +51,7 @@ var OP;
     OP[OP["TANH"] = 43] = "TANH";
     OP[OP["T"] = 44] = "T";
     OP[OP["MM"] = 45] = "MM";
+    OP[OP["DOT"] = 46] = "DOT";
 })(OP || (exports.OP = OP = {}));
 class Node {
     value;
@@ -447,6 +448,15 @@ class Node {
         };
         return out;
     }
+    dot(other) {
+        other = Node.forceNode(other);
+        const out = new Node(dot(this.value, other.value), [this, other], OP.DOT);
+        out.feedBackward = () => {
+            Node.addGrad(this, mul(out.grad, other.value));
+            Node.addGrad(other, mul(out.grad, this.value));
+        };
+        return out;
+    }
     backward() {
         // Build topological order
         const topo = [];

package/dist/tensor.d.ts

@@ -54,4 +54,5 @@ export declare class TensorMath {
     static sum(tA: Tensor, dims?: number[] | number, keepDims?: boolean): Tensor;
     static t(tA: Tensor): Tensor;
     static mm(tA: Tensor, tB: Tensor): Tensor;
+    static dot(tA: Tensor, tB: Tensor): Tensor;
 }

package/dist/tensor.js

@@ -289,5 +289,19 @@ class TensorMath {
         }
         return matC;
     }
+    static dot(tA, tB) {
+        const shapeA = TensorMath.getShape(tA);
+        const shapeB = TensorMath.getShape(tB);
+        if (shapeA.length !== 1 || shapeB.length !== 1 || shapeA[0] !== shapeB[0])
+            throw new Error("Inputs are not 1D tensors");
+        const vectLen = shapeA[0];
+        const vectA = tA;
+        const vectB = tB;
+        let sum = 0;
+        for (let index = 0; index < vectLen; index++) {
+            sum += vectA[index] * vectB[index];
+        }
+        return sum;
+    }
 }
 exports.TensorMath = TensorMath;

package/package.json

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