npm - catniff - Versions diffs - 0.6.5 → 0.6.6 - Mend

catniff 0.6.5 → 0.6.6

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (2) hide show

package/dist/core.js +11 -2
package/package.json +1 -1

package/dist/core.js CHANGED Viewed

@@ -337,7 +337,12 @@ class Tensor {
             throw new Error("Can not create view: incompatible metadata");
         }
         const outputStrides = Tensor.getStrides(newShape);
-        const out = new Tensor(this.value, { shape: newShape, strides: outputStrides, numel: outputSize });
+        const out = new Tensor(this.value, {
+            shape: newShape,
+            strides: outputStrides,
+            numel: outputSize,
+            device: this.device
+        });
         // Gradient reshaped and flow back to the original tensor
         if (this.requiresGrad) {
             out.requiresGrad = true;
@@ -357,7 +362,11 @@ class Tensor {
         }
         // Create new tensor with forced compatibility (only contiguity for now)
         const outputStrides = Tensor.getStrides(newShape);
-        const out = new Tensor(this.contiguous().value, { shape: newShape, strides: outputStrides, numel: outputSize });
+        const out = new Tensor(this.contiguous().value, {
+            shape: newShape,
+            strides: outputStrides,
+            numel: outputSize
+        });
         // Gradient reshaped and flow back to the original tensor
         if (this.requiresGrad) {
             out.requiresGrad = true;

package/package.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "name": "catniff",
-  "version": "0.6.5",
+  "version": "0.6.6",
   "description": "A small Torch-like deep learning framework for Javascript",
   "main": "index.js",
   "scripts": {