fix

deepjavalibrary · Jul 2, 2023 · f9eef46 · f9eef46
1 parent 78ddcef
commit f9eef46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/utils/DataPoints.java b/utils/DataPoints.java
@@ -24,7 +24,7 @@ public NDArray getY() {
     // Generate y = X w + b + noise
     public static DataPoints syntheticData(NDManager manager, NDArray w, float b, int numExamples) {
         NDArray X = manager.randomNormal(new Shape(numExamples, w.size()));
-        NDArray y = X.dot(w).add(b);
+        NDArray y = X.matMul(w).add(b);
         // Add noise
         y = y.add(manager.randomNormal(0, 0.01f, y.getShape(), DataType.FLOAT32));
         return new DataPoints(X, y);

diff --git a/utils/Training.java b/utils/Training.java
@@ -17,7 +17,7 @@
 class Training {
 
     public static NDArray linreg(NDArray X, NDArray w, NDArray b) {
-        return X.dot(w).add(b);
+        return X.matMul(w).add(b);
     }
 
     public static NDArray squaredLoss(NDArray yHat, NDArray y) {