Add linspace and linspace's unittest.

2023-04-26 13:41:57 +08:00
parent 7693ce9f9e
commit 6fd22ffc89
3 changed files with 25 additions and 0 deletions
--- a/src/Function1D.cpp
+++ b/src/Function1D.cpp
@@ -538,3 +538,15 @@ Matrix Aurora::vecnorm(const Matrix& aMatrix, NormMethod aNormMethod, int aDim)

    return Matrix::New(resultData,column);
 }
+
+Matrix Aurora::linspace(double aStart, double aEnd, int aNum)
+{
+    double step = (aEnd - aStart) / (aNum - 1);
+    double* resultData = Aurora::malloc(aNum);
+    for (int i = 0; i < aNum; i++)
+    {
+        resultData[i] = aStart + step * i;
+    }
+
+    return Matrix::New(resultData,aNum);
+}