fix: removeDataFromArrays while removelist equal zero.

This commit is contained in:
sunwen
2025-10-22 10:02:13 +08:00
parent 1435dcb94e
commit 1b1450e85c

View File

@@ -13,6 +13,11 @@ using namespace Aurora;
Aurora::Matrix Recon::removeDataFromArrays(const Aurora::Matrix& aOutPutMatrix, const Aurora::Matrix& aRemoveList)
{
if(Aurora::sum(aRemoveList)[0] == 0)
{
return aOutPutMatrix;
}
size_t columnSize = sum(aRemoveList, FunctionDirection::All)[0];
size_t rowSize = aOutPutMatrix.getDimSize(0);
float* resultData = Aurora::malloc(columnSize * rowSize);