catniff 0.2.8 → 0.2.9

package/dist/core.d.ts

@@ -26,7 +26,7 @@ export declare class Tensor {
         readonly number[]
     ];
     static broadcastShapes(shapeA: readonly number[], shapeB: readonly number[]): readonly number[];
-    static indexToCoords(index: number, shape: readonly number[], strides: readonly number[]): number[];
+    static indexToCoords(index: number, strides: readonly number[]): number[];
     static coordsToUnbroadcastedIndex(coords: number[], shape: readonly number[], strides: readonly number[]): number;
     static coordsToIndex(coords: number[], strides: readonly number[]): number;
     static shapeToSize(shape: readonly number[]): number;

package/dist/core.js

@@ -94,12 +94,10 @@ class Tensor {
         return newShape;
     }
     // Utility to convert flat index to array of coordinates
-    static indexToCoords(index, shape, strides) {
-        const coords = new Array(shape.length);
+    static indexToCoords(index, strides) {
+        const coords = new Array(strides.length);
         let remaining = index;
-        // Sort dimensions by stride (largest first) for correct decomposition
-        const sortedDims = shape.map((_, i) => i).sort((a, b) => strides[b] - strides[a]);
-        for (const dim of sortedDims) {
+        for (let dim = 0; dim < strides.length; dim++) {
             coords[dim] = Math.floor(remaining / strides[dim]);
             remaining %= strides[dim];
         }
@@ -152,7 +150,7 @@ class Tensor {
         const outputValue = new Array(outputSize);
         for (let i = 0; i < outputSize; i++) {
             // Get coordinates from 1D index
-            const coordsOutput = Tensor.indexToCoords(i, outputShape, outputStrides);
+            const coordsOutput = Tensor.indexToCoords(i, outputStrides);
             // Convert the coordinates to 1D index of flattened A with respect to A's shape
             const indexA = Tensor.coordsToUnbroadcastedIndex(coordsOutput, paddedAShape, paddedAStrides);
             // Convert the coordinates to 1D index of flattened B with respect to B's shape
@@ -267,14 +265,19 @@ class Tensor {
             }
         }
         // Remove size-1 dims only
-        const outShape = this.shape.filter((dim, i) => {
-            const shouldSqueeze = dims.includes(i);
-            if (shouldSqueeze && dim !== 1)
-                throw new Error(`Can not squeeze dim with size ${dim}`);
-            return !shouldSqueeze;
-        });
-        // Remove strides of size-1 dims
-        const outStrides = this.strides.filter((stride, i) => !dims.includes(i));
+        const outShape = [], outStrides = [];
+        for (let index = 0; index < this.shape.length; index++) {
+            const dim = this.shape[index];
+            const stride = this.strides[index];
+            if (dims.includes(index)) {
+                if (dim !== 1)
+                    throw new Error(`Can not squeeze dim with size ${dim}`);
+            }
+            else {
+                outShape.push(dim);
+                outStrides.push(stride);
+            }
+        }
         const outValue = outShape.length === 0 ? this.value[0] : this.value;
         const out = new Tensor(outValue, {
             shape: outShape,
@@ -353,7 +356,7 @@ class Tensor {
         }
         // Calculate new value after sum
         for (let index = 0; index < originalSize; index++) {
-            const coords = Tensor.indexToCoords(index, this.shape, this.strides);
+            const coords = Tensor.indexToCoords(index, this.strides);
             // Force 0 on reduced axes to collapse into size-1 dims
             const outCoords = coords.map((val, i) => dims.includes(i) ? 0 : val);
             // Convert output coordinates to flat index
@@ -407,7 +410,7 @@ class Tensor {
         }
         // Calculate new value after multiplying
         for (let index = 0; index < originalSize; index++) {
-            const coords = Tensor.indexToCoords(index, this.shape, this.strides);
+            const coords = Tensor.indexToCoords(index, this.strides);
             // Force 0 on reduced axes to collapse into size-1 dims
             const outCoords = coords.map((val, i) => dims.includes(i) ? 0 : val);
             // Convert output coordinates to flat index
@@ -424,7 +427,7 @@ class Tensor {
         if (this.requiresGrad) {
             // Grad is the product of other elements of the same axis, which is product of all els divided by the current value
             for (let index = 0; index < originalSize; index++) {
-                const coords = Tensor.indexToCoords(index, this.shape, this.strides);
+                const coords = Tensor.indexToCoords(index, this.strides);
                 // Force 0 on reduced axes to collapse into size-1 dims
                 const outCoords = coords.map((val, i) => dims.includes(i) ? 0 : val);
                 // Convert output coordinates to flat index
@@ -470,7 +473,7 @@ class Tensor {
         }
         // Calculate sums and how many elements contribute to specific positions
         for (let index = 0; index < originalSize; index++) {
-            const coords = Tensor.indexToCoords(index, this.shape, this.strides);
+            const coords = Tensor.indexToCoords(index, this.strides);
             // Force 0 on reduced axes to collapse into size-1 dims
             const outCoords = coords.map((val, i) => dims.includes(i) ? 0 : val);
             // Convert output coordinates to flat index
@@ -492,7 +495,7 @@ class Tensor {
         if (this.requiresGrad) {
             // Calculate grad by assiging 1 divide by the number of contributors to the position
             for (let index = 0; index < originalSize; index++) {
-                const coords = Tensor.indexToCoords(index, this.shape, this.strides);
+                const coords = Tensor.indexToCoords(index, this.strides);
                 // Force 0 on reduced axes to collapse into size-1 dims
                 const outCoords = coords.map((val, i) => dims.includes(i) ? 0 : val);
                 // Convert output coordinates to flat index

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "catniff",
-  "version": "0.2.8",
+  "version": "0.2.9",
   "description": "A cute autograd engine for Javascript",
   "main": "index.js",
   "scripts": {
@@ -26,7 +26,9 @@
     "neural-network",
     "machine-learning",
     "deep-learning",
-    "micrograd"
+    "micrograd",
+    "torch",
+    "pytorch"
   ],
   "author": "nguyenphuminh",
   "license": "GPL-3.0",