提交 9367e589 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3697 from Sentient07/issue-3687

Testing the output shape
......@@ -195,6 +195,10 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
maxpool_op = DownsampleFactorMax(maxpoolshp,
ignore_border=ignore_border,
mode=mode)(images)
output_shape = DownsampleFactorMax.out_shape(imval.shape, maxpoolshp,
ignore_border=ignore_border)
utt.assert_allclose(numpy.asarray(output_shape), numpy_output_val.shape)
f = function([images], maxpool_op)
output_val = f(imval)
utt.assert_allclose(output_val, numpy_output_val)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论