From f9eef460666ece766e9f3786287fc242411cb0c0 Mon Sep 17 00:00:00 2001
From: KexinFeng <fenkexin@amazon.com>
Date: Sun, 2 Jul 2023 11:26:52 -0700
Subject: [PATCH] fix

---
 utils/DataPoints.java | 2 +-
 utils/Training.java   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils/DataPoints.java b/utils/DataPoints.java
index 1db60668..6c9dfa5e 100644
--- 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
index e1e46508..20fa4f77 100644
--- 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) {