提交 06dbf2a9 authored 作者: Ramana.S's avatar Ramana.S

Test passes

上级 9c6627b9
......@@ -196,13 +196,9 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
ignore_border=ignore_border,
mode=mode)(images)
output_shape = DownsampleFactorMax.out_shape(imval, maxpoolshp,
output_shape = DownsampleFactorMax.out_shape(imval.shape, maxpoolshp,
ignore_border=ignore_border)
assert numpy.asarray(output_shape).shape == numpy_output_val.shape, (
"outshape is %s, calculated shape is %s"
% (numpy.asarray(output_shape).shape, numpy_output_val.shape))
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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论