catniff 0.5.8 → 0.5.9

package/dist/backend.d.ts

@@ -1,4 +1,5 @@
 import { Tensor } from "./core";
 export interface Backend {
+    create(tensor: Tensor): void;
     transfer(tensor: Tensor): Tensor;
 }

package/dist/core.d.ts

@@ -177,4 +177,5 @@ export declare class Tensor {
     replace(other: Tensor, allowShapeMismatch?: boolean): Tensor;
     static backends: Map<string, Backend>;
     to(device: string): Tensor;
+    to_(device: string): Tensor;
 }

package/dist/core.js

@@ -21,13 +21,8 @@ class Tensor {
         this.gradFn = options.gradFn || (() => { });
         this.children = options.children || [];
         this.device = options.device || "cpu";
-        // Move tensor to device
-        if (this.device !== "cpu") {
-            const backend = Tensor.backends.get(this.device);
-            if (backend && backend.transfer) {
-                backend.transfer(this);
-            }
-        }
+        // Move to device in-place
+        this.to_(this.device);
     }
     // Utility to flatten an nD array to be 1D
     static flatten(tensor) {
@@ -1712,9 +1707,21 @@ class Tensor {
     static backends = new Map();
     // Op to transfer tensor to another device
     to(device) {
+        if (device === "cpu")
+            return this;
         const backend = Tensor.backends.get(device);
         if (backend && backend.transfer) {
-            backend.transfer(this);
+            return backend.transfer(this);
+        }
+        throw new Error(`No device found to transfer tensor to or a handler is not implemented for device.`);
+    }
+    // Op to transfer tensor to another device in-place
+    to_(device) {
+        if (device === "cpu")
+            return this;
+        const backend = Tensor.backends.get(this.device);
+        if (backend && backend.create) {
+            backend.create(this);
             return this;
         }
         throw new Error(`No device found to transfer tensor to or a handler is not implemented for device.`);

package/package.json

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