cherry-pick dot fixes from en branch (#28)

deepjavalibrary · Jul 5, 2023 · 2b653fd · 2b653fd
1 parent d062e55
commit 2b653fd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/chapter_linear-networks/linear-regression-scratch.ipynb b/chapter_linear-networks/linear-regression-scratch.ipynb
@@ -70,7 +70,7 @@
     "// Generate y = X w + b + noise\n",
     "public DataPoints syntheticData(NDManager manager, NDArray w, float b, int numExamples) {\n",
     "    NDArray X = manager.randomNormal(new Shape(numExamples, w.size()));\n",
-    "    NDArray y = X.dot(w).add(b);\n",
+    "    NDArray y = X.matMul(w).add(b);\n",
     "    // Add noise\n",
     "    y = y.add(manager.randomNormal(0, 0.01f, y.getShape(), DataType.FLOAT32));\n",
     "    return new DataPoints(X, y);\n",

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
@@ -16,7 +16,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) {