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

Add functions to 1D(nan, finite, padding, not)

Browse Source
This commit is contained in:
kradchen
2023-11-28 14:26:46 +08:00
parent 1716b139c4
commit 3438121b0c
4 changed files with 178 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/Function1D_Test.cpp
View File

@@ -560,8 +560,10 @@ TEST_F(Function1D_Test, padding) {
float *input = new float[18]{10,2,1,3,4,4,16,3,1,2,15,-2,1,-3,4,-4,1,-3};
auto ma = Aurora::Matrix::fromRawData(input,18,1,1);
Aurora::padding(ma,20,1);
auto result = ma.getData();
EXPECT_FLOAT_AE(ma[19],1);
EXPECT_FLOAT_AE(ma[20],1);
Aurora::padding(ma,10,-1);
EXPECT_FLOAT_AE(ma[10],-1);
}
TEST_F(Function1D_Test, convertfp16tofloat) {