sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix ones and zeros bug.

Browse Source
This commit is contained in:
kradchen
2023-12-18 13:47:38 +08:00
parent 6dfea91ed0
commit b09257298c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/Function3D.cpp
View File

@@ -66,7 +66,7 @@ Matrix Aurora::ones(int aRow, int aColumn, int aSlice) {
size_t arraySize = rowSize * colSize* sliceSize; size_t arraySize = rowSize * colSize* sliceSize;
float* data = malloc(arraySize); float* data = malloc(arraySize);
float one = 1.0; float one = 1.0;
cblas_scopy(arraySize,&one,0,data,1.0f); cblas_scopy(arraySize,&one,0,data,1.0);
return Matrix::New(data,rowSize,colSize,aSlice); return Matrix::New(data,rowSize,colSize,aSlice);
} }
@@ -106,7 +106,7 @@ Matrix Aurora::zeros(int aRow, int aColumn, int aSlice) {
size_t arraySize = rowSize * colSize* sliceSize; size_t arraySize = rowSize * colSize* sliceSize;
float* data = malloc(arraySize); float* data = malloc(arraySize);
float zero = 0.0; float zero = 0.0;
cblas_scopy(arraySize,&zero,0,data,0.0f); cblas_scopy(arraySize,&zero,0,data,1);
return Matrix::New(data,rowSize,colSize,sliceSize); return Matrix::New(data,rowSize,colSize,sliceSize);
} }