catniff 0.8.9 → 0.8.10

package/dist/core.d.ts

@@ -101,6 +101,8 @@ export declare class Tensor {
     softmin(dim?: number): Tensor;
     logsumexp(dim?: number): Tensor;
     logSoftmax(dim?: number): Tensor;
+    logaddexp(other: TensorValue | Tensor): Tensor;
+    lerp(end: TensorValue | Tensor, weight: TensorValue | Tensor): Tensor;
     add(other: TensorValue | Tensor): Tensor;
     sub(other: TensorValue | Tensor): Tensor;
     subtract: (other: TensorValue | Tensor) => Tensor;

package/dist/core.js

@@ -1272,6 +1272,20 @@ class Tensor {
         }
         return this.sub(this.logsumexp(dim));
     }
+    // Tensor (stable) logaddexp
+    logaddexp(other) {
+        other = this.handleOther(other);
+        const max = this.maximum(other);
+        const exp1 = this.sub(max).exp();
+        const exp2 = other.sub(max).exp();
+        return max.add(exp1.add(exp2).log());
+    }
+    // Tensor linear interpolation
+    lerp(end, weight) {
+        end = this.handleOther(end);
+        weight = this.handleOther(weight);
+        return this.add(weight.mul(end.sub(this)));
+    }
     // Tensor element-wise addition
     add(other) {
         return this.elementWiseABDAG(other, (a, b) => a + b, (self, other, outGrad) => outGrad, (self, other, outGrad) => outGrad);

package/package.json

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