Fix radius calculation error in Tensor.add method

ProbableTrain · May 3, 2020 · efb0275 · efb0275
1 parent bb2d544
commit efb0275
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ts/impl/tensor.ts b/src/ts/impl/tensor.ts
@@ -39,7 +39,8 @@ export default class Tensor {
 
     add(tensor: Tensor): Tensor {
         this.matrix = this.matrix.map((v, i) => v * this.r + tensor.matrix[i] * tensor.r);
-        this.r = 2;
+        this.r = Math.hypot(...this.matrix);
+        this.matrix = this.matrix.map(v => v / this.r);
         this.oldTheta = true;
         return this;
     }