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

Add slice support to horncat and vertcat

Browse Source
This commit is contained in:
kradchen
2023-06-09 16:56:56 +08:00
parent 428d75e548
commit 461c856bd9
2 changed files with 30 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/Function1D_Test.cpp
View File

@@ -429,6 +429,18 @@ TEST_F(Function1D_Test, horzcat) {
EXPECT_DOUBLE_EQ(result.getData()[9],6);
EXPECT_DOUBLE_EQ(result.getDimSize(0),3);
EXPECT_DOUBLE_EQ(result.getDimSize(1),2);
auto A = Aurora::Matrix::New(Aurora::random(27),3,3,3);
auto B = Aurora::Matrix::New(Aurora::random(18),3,2,3);
auto C = Aurora::horzcat(A, B);
A.printf();
B.printf();
C.printf();
B = Aurora::Matrix::New(Aurora::random(18),2,3,3);
C = Aurora::vertcat(A, B);
A.printf();
B.printf();
C.printf();
}
TEST_F(Function1D_Test, vertcat) {