提交 e3c121c8 authored 作者: Frederic Bastien's avatar Frederic Bastien

Add test for the regression fix

上级 86e3535a
......@@ -25,6 +25,10 @@ from theano import function
class TestDownsampleFactorMax(utt.InferShapeTester):
def test_out_shape(self):
assert Pool.out_shape((9, 8, 6), (2, 2)) == [9, 4, 3]
assert Pool.out_shape((8, 6), (2, 2)) == [4, 3]
@staticmethod
def numpy_max_pool_2d(input, ds, ignore_border=False, mode='max'):
'''Helper function, implementing pool_2d in pure numpy'''
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论