提交 b1546706 authored 作者: Alexander Matyasko's avatar Alexander Matyasko

Add tests for 6d and 7d input

上级 85022b36
...@@ -45,7 +45,10 @@ def test_pool2d(): ...@@ -45,7 +45,10 @@ def test_pool2d():
(1, 1, 10, 1025), (1, 1, 10, 1025),
(1, 1, 10, 1023), (1, 1, 10, 1023),
(1, 1, 1025, 10), (1, 1, 1025, 10),
(1, 1, 1023, 10), ] (1, 1, 1023, 10),
(3, 2, 16, 16, 16),
(3, 2, 6, 6, 6, 5),
(3, 2, 6, 6, 6, 5, 7), ]
numpy.random.RandomState(utt.fetch_seed()).shuffle(shps) numpy.random.RandomState(utt.fetch_seed()).shuffle(shps)
test_ws = (2, 2), (3, 2), (1, 1) test_ws = (2, 2), (3, 2), (1, 1)
...@@ -138,7 +141,9 @@ def test_pool3d(): ...@@ -138,7 +141,9 @@ def test_pool3d():
(1, 1, 10, 10, 1025), (1, 1, 10, 10, 1025),
(1, 1, 10, 10, 1023), (1, 1, 10, 10, 1023),
(1, 1, 10, 1025, 10), (1, 1, 10, 1025, 10),
(1, 1, 10, 1023, 10), ] (1, 1, 10, 1023, 10),
(3, 2, 6, 6, 6, 5),
(3, 2, 6, 6, 6, 5, 7), ]
numpy.random.RandomState(utt.fetch_seed()).shuffle(shps) numpy.random.RandomState(utt.fetch_seed()).shuffle(shps)
test_ws = (2, 2, 2), (3, 2, 3), (1, 1, 1) test_ws = (2, 2, 2), (3, 2, 3), (1, 1, 1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论