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 *)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 *)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)); } }