From c71e359ce5826ddd2ce4c1836eea4900e0c74194 Mon Sep 17 00:00:00 2001
From: Krad <kradchen123@163.com>
Date: Tue, 25 Apr 2023 17:37:22 +0800
Subject: [PATCH] Fix abs bug

---
 src/Function1D.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Function1D.cpp b/src/Function1D.cpp
index 8d01ad4..1fb6a5c 100644
--- a/src/Function1D.cpp
+++ b/src/Function1D.cpp
@@ -128,7 +128,7 @@ Aurora::Matrix Aurora::abs(const Aurora::Matrix &matrix) {
     else{
         vzAbsI(matrix.getDataSize(), (std::complex<double> *)matrix.getData(), SAME_STRIDE,output, SAME_STRIDE);
     }
-    return Aurora::Matrix::New(output, matrix);
+    return Aurora::Matrix::New(output, matrix.getDimSize(0), matrix.getDimSize(1), matrix.getDimSize(2));
 }
 
 Aurora::Matrix Aurora::abs(Aurora::Matrix&& matrix) {
@@ -140,7 +140,7 @@ Aurora::Matrix Aurora::abs(Aurora::Matrix&& matrix) {
     else{
         auto output = malloc(matrix.getDataSize());
         vzAbsI(matrix.getDataSize(), (std::complex<double> *)matrix.getData(), SAME_STRIDE,output, SAME_STRIDE);
-        return Aurora::Matrix::New(output, matrix);
+        return Aurora::Matrix::New(output, matrix.getDimSize(0), matrix.getDimSize(1), matrix.getDimSize(2));
     }
 
 }