catniff 0.6.7 → 0.6.8

package/dist/core.d.ts

@@ -57,6 +57,7 @@ export declare class Tensor {
     indexWithArray(indices: number[]): Tensor;
     index(indices: Tensor | TensorValue): Tensor;
     slice(ranges: number[][]): Tensor;
+    chunk(chunks: number, dim?: number): Tensor[];
     squeeze(dims?: number[] | number): Tensor;
     unsqueeze(dim: number): Tensor;
     static reduce(tensor: Tensor, dims: number[] | number | undefined, keepDims: boolean, config: {

package/dist/core.js

@@ -605,6 +605,25 @@ class Tensor {
         }
         return out;
     }
+    // Tensor chunk
+    chunk(chunks, dim = 0) {
+        // Handle negative indices
+        if (dim < 0) {
+            dim += this.shape.length;
+        }
+        const sliceOpt = new Array(this.shape.length);
+        for (let index = 0; index < sliceOpt.length; index++) {
+            sliceOpt[index] = [];
+        }
+        const dimSize = this.shape[dim];
+        const chunkDimSize = Math.ceil(dimSize / chunks);
+        const results = [];
+        for (let index = 0; index < dimSize; index += chunkDimSize) {
+            sliceOpt[dim] = [index, Math.min(index + chunkDimSize, dimSize)];
+            results.push(this.slice(sliceOpt));
+        }
+        return results;
+    }
     // Tensor squeeze
     squeeze(dims) {
         if (typeof this.value === "number")

package/package.json

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