catniff 0.9.0 → 0.9.1

package/dist/core.d.ts

@@ -202,6 +202,7 @@ export declare class Tensor {
     gelu(approximate?: string): Tensor;
     maximum(other: TensorValue | Tensor): Tensor;
     minimum(other: TensorValue | Tensor): Tensor;
+    copysign(other: TensorValue | Tensor): Tensor;
     round(): Tensor;
     floor(): Tensor;
     ceil(): Tensor;

package/dist/core.js

@@ -328,14 +328,9 @@ class Tensor {
         }
         const reducedGrad = accumGrad.sum(axesToReduce, true);
         const squeezedGrad = reducedGrad.squeeze(axesToSqueeze);
-        // Enforce 0-offset contiguous grads and correct dtype
         if (typeof tensor.grad === "undefined") {
-            let grad = squeezedGrad;
-            // Handle potentially contiguous tensors with non zero offset
-            if (grad.offset !== 0) {
-                grad = grad.clone();
-            }
-            tensor.grad = grad.contiguous().cast(tensor.dtype);
+            // Force default grad to have same shape and dtype as original tensor
+            tensor.grad = Tensor.zerosLike(tensor).add(squeezedGrad.cast(tensor.dtype));
         }
         else {
             tensor.grad = tensor.grad.add(squeezedGrad.cast(tensor.dtype));
@@ -1815,6 +1810,10 @@ class Tensor {
     minimum(other) {
         return this.elementWiseABDAG(other, (a, b) => Math.min(a, b), (self, other, outGrad) => outGrad.mul(self.lt(other).add(self.eq(other).mul(0.5))), (self, other, outGrad) => outGrad.mul(other.lt(self).add(other.eq(self).mul(0.5))));
     }
+    // Tensor element-wise copysign
+    copysign(other) {
+        return this.elementWiseABDAG(other, (a, b) => Math.abs(a) * (Object.is(b, -0) || b < 0 ? -1 : 1), (self, other, outGrad) => outGrad.mul(self.sign().mul(other.sign())), (self, other, outGrad) => new Tensor(0));
+    }
     // Tensor element-wise round
     round() {
         return this.elementWiseSelfDAG((a) => Math.round(a));
@@ -2235,7 +2234,7 @@ class Tensor {
         else {
             // Each group handles Cin/groups input channels and Cout/groups output channels.
             // chunk(groups, 2) splits the Cin*kH*kW axis into groups equal slices,
-            // each of size CinPerGroup*kH*kW — valid because reshape laid Cin outermost.
+            // each of size CinPerGroup*kH*kW - valid because reshape laid Cin outermost.
             const patchChunks = x.chunk(groups, 2); // Tensor[groups], each [N, Hout*Wout, CinPerGroup*kH*kW]
             const weightChunks = w.chunk(groups, 0); // Tensor[groups], each [Cout/groups, CinPerGroup*kH*kW]
             const groupOuts = patchChunks.map((patch, i) => patch.matmul(weightChunks[i].t()) // [N, Hout*Wout, Cout/groups]

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "catniff",
-  "version": "0.9.0",
+  "version": "0.9.1",
   "description": "Torch-like deep learning framework for Javascript",
   "main": "./dist/index.js",
   "scripts": {