Add exp and exp's unittest.

This commit is contained in:
sunwen
2023-04-23 15:47:29 +08:00
parent 3833040672
commit 9212115e47
4 changed files with 39 additions and 3 deletions

View File

@@ -333,6 +333,22 @@ Matrix Aurora::log(const Matrix& aMatrix, int aBaseNum)
data[i] /= temp;
}
}
return Aurora::Matrix(std::shared_ptr<double>(data,Aurora::free),std::vector<int>{aMatrix.getDimSize(0),aMatrix.getDimSize(1),aMatrix.getDimSize(2)});
return Matrix::New(data, aMatrix);
}
Matrix Aurora::exp(const Matrix& aMatrix)
{
size_t size = aMatrix.getDataSize();
double* data;
if (aMatrix.isComplex())
{
data = Aurora::malloc(size, true);
vzExp(size, (MKL_Complex16*)aMatrix.getData(), (MKL_Complex16*)data);
}
else
{
data = Aurora::malloc(size);
vdExp(size, aMatrix.getData(), data);
}
return Matrix::New(data, aMatrix);
}