Fix function by default value 1 in matrix mInfo .

This commit is contained in:
sunwen
2023-04-21 14:42:24 +08:00
parent a09a748051
commit 0c8e1e1038
3 changed files with 4 additions and 16 deletions

View File

@@ -241,12 +241,7 @@ Matrix Aurora::repmat(const Matrix& aMatrix,int aRowTimes, int aColumnTimes)
int originalDataSize = aMatrix.getDataSize() * complexStep;
double* resultData = Aurora::malloc(originalDataSize * aRowTimes * aColumnTimes);
int row = aMatrix.getDimSize(0);
int column = 1;
if(aMatrix.getDims() > 1)
{
column = aMatrix.getDimSize(1);
}
int column = aMatrix.getDimSize(1);
double* originalData = aMatrix.getData();
double* resultDataTemp = resultData;
for(int i=0; i<column; ++i)
@@ -295,11 +290,7 @@ Matrix Aurora::repmat(const Matrix& aMatrix,int aRowTimes, int aColumnTimes, int
}
std::vector<int> resultInfo;
int row = resultTemp.getDimSize(0);
int column = 1;
if(resultTemp.getDims() > 1)
{
column = resultTemp.getDimSize(1);
}
int column = resultTemp.getDimSize(1);
resultInfo.push_back(row);
if (column > 1 || aSliceTimes > 1)
{