catniff 0.7.2 → 0.7.4

package/dist/core.d.ts

@@ -61,6 +61,7 @@ export declare class Tensor {
     index(indices: Tensor | TensorValue): Tensor;
     slice(ranges: number[][]): Tensor;
     chunk(chunks: number, dim?: number): Tensor[];
+    expand(newShape: number[]): Tensor;
     cat(other: Tensor | TensorValue, dim?: number): Tensor;
     squeeze(dims?: number[] | number): Tensor;
     unsqueeze(dim: number): Tensor;
@@ -219,7 +220,7 @@ export declare class Tensor {
     val(): TensorValue;
     detach(): Tensor;
     clone(): Tensor;
-    replace(other: Tensor | TensorValue, allowShapeMismatch?: boolean): Tensor;
+    replace(other: Tensor | TensorValue): Tensor;
     static backends: Map<string, Backend>;
     to(device: string): Tensor;
     to_(device: string): Tensor;

package/dist/core.js

@@ -647,6 +647,46 @@ class Tensor {
         }
         return results;
     }
+    // Tensor expansion
+    expand(newShape) {
+        // Handle scalars
+        let self = this;
+        if (typeof this.value === "number") {
+            self = self.unsqueeze(0);
+        }
+        // Pad shapes to same length
+        const ndim = Math.max(self.shape.length, newShape.length);
+        const oldShape = [...Array(ndim - self.shape.length).fill(1), ...self.shape];
+        const oldStrides = [...Array(ndim - self.strides.length).fill(0), ...self.strides];
+        const targetShape = [...Array(ndim - newShape.length).fill(1), ...newShape];
+        const newStrides = new Array(ndim);
+        for (let i = 0; i < ndim; i++) {
+            if (oldShape[i] === targetShape[i]) {
+                newStrides[i] = oldStrides[i];
+            }
+            else if (oldShape[i] === 1) {
+                newStrides[i] = 0;
+            }
+            else {
+                throw new Error(`Cannot expand dimension of size ${oldShape[i]} to ${targetShape[i]}`);
+            }
+        }
+        const out = new Tensor(self.value, {
+            shape: targetShape,
+            strides: newStrides,
+            offset: self.offset,
+            numel: Tensor.shapeToSize(targetShape),
+            device: self.device
+        });
+        if (self.requiresGrad) {
+            out.requiresGrad = true;
+            out.children.push(self);
+            out.gradFn = () => {
+                Tensor.addGrad(self, out.grad);
+            };
+        }
+        return out;
+    }
     // Tensor concatentation
     cat(other, dim = 0) {
         other = this.handleOther(other);
@@ -1954,17 +1994,21 @@ class Tensor {
         return out;
     }
     // Returns this tensor with value replaced with the value of another tensor
-    replace(other, allowShapeMismatch = false) {
+    replace(other) {
         other = this.handleOther(other);
         // Verify shape
-        if (!allowShapeMismatch) {
-            for (let index = 0; index < this.shape.length; index++) {
-                if (this.shape[index] !== other.shape[index]) {
-                    throw new Error("Shape mismatch when trying to do tensor value replacement");
-                }
+        if (this.shape.length !== other.shape.length) {
+            throw new Error("Shape mismatch when trying to do tensor value replacement");
+        }
+        for (let index = 0; index < this.shape.length; index++) {
+            if (this.shape[index] !== other.shape[index]) {
+                throw new Error("Shape mismatch when trying to do tensor value replacement");
             }
         }
+        // Reassign values
         this.value = other.value;
+        this.strides = other.strides;
+        this.offset = other.offset;
         return this;
     }
     // Holds all available backends

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "catniff",
-  "version": "0.7.2",
-  "description": "A small Torch-like deep learning framework for Javascript",
+  "version": "0.7.4",
+  "description": "Torch-like deep learning framework for Javascript",
   "main": "index.js",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"