Browse Source

more slice test

tags/20210124
nihui 5 years ago
parent
commit
80fdddb502
1 changed files with 23 additions and 10 deletions
  1. +23
    -10
      tests/test_slice.cpp

+ 23
- 10
tests/test_slice.cpp View File

@@ -93,10 +93,14 @@ static int test_slice_0()
static int test_slice_1()
{
ncnn::Mat a = RandomMat(7, 3, 16);
ncnn::Mat b = RandomMat(5, 4, 20);

return 0
|| test_slice(a, IntArrayMat(3, 8, -233), 0)
|| test_slice(a, IntArrayMat(3, 8, -233), -3);
|| test_slice(a, IntArrayMat(3, 8, -233), -3)

|| test_slice(b, IntArrayMat(4, -233), 0)
|| test_slice(b, IntArrayMat(4, -233), -3);
}

static int test_slice_2()
@@ -128,31 +132,40 @@ static int test_slice_3()
static int test_slice_4()
{
ncnn::Mat a = RandomMat(7, 16);
ncnn::Mat b = RandomMat(16, 2);
ncnn::Mat c = RandomMat(16, 8);
ncnn::Mat b = RandomMat(5, 20);

ncnn::Mat c = RandomMat(15, 2);
ncnn::Mat d = RandomMat(16, 8);

return 0
|| test_slice(a, IntArrayMat(3, 8, 5), 0)
|| test_slice(a, IntArrayMat(3, 8, 5), -2)

|| test_slice(b, IntArrayMat(3, -233, -233), 1)
|| test_slice(b, IntArrayMat(3, -233, -233), -1)
|| test_slice(b, IntArrayMat(4, -233), 0)
|| test_slice(b, IntArrayMat(4, -233), -2)

|| test_slice(c, IntArrayMat(3, 8, 5), 1)
|| test_slice(c, IntArrayMat(3, 8, 5), -1);
|| test_slice(c, IntArrayMat(3, -233, -233), 1)
|| test_slice(c, IntArrayMat(3, -233, -233), -1)

|| test_slice(d, IntArrayMat(3, 8, 5), 1)
|| test_slice(d, IntArrayMat(3, 8, 5), -1);
}

static int test_slice_5()
{
ncnn::Mat a = RandomMat(16);
ncnn::Mat b = RandomMat(24);
ncnn::Mat b = RandomMat(20);
ncnn::Mat c = RandomMat(24);

return 0
|| test_slice(a, IntArrayMat(3, 8, 5), 0)
|| test_slice(a, IntArrayMat(3, 8, 5), -1)

|| test_slice(b, IntArrayMat(4, 8, -233), 0)
|| test_slice(b, IntArrayMat(4, 8, -233), -1);
|| test_slice(b, IntArrayMat(4, -233), 0)
|| test_slice(b, IntArrayMat(4, -233), -1)

|| test_slice(c, IntArrayMat(4, 8, -233), 0)
|| test_slice(c, IntArrayMat(4, 8, -233), -1);
}

int main()


Loading…
Cancel
Save