catniff 0.7.3 → 0.7.4

package/dist/core.d.ts

@@ -220,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

@@ -1994,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.3",
-  "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"