提交 8590f4a7 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Use assert_allclose to check downsample result

上级 104eb5fa
......@@ -189,7 +189,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
mode=mode)
f = function([images, ], [output, ])
output_val = f(imval)
assert numpy.all(output_val == numpy_output_val)
utt.assert_allclose(output_val, numpy_output_val)
# DownsampleFactorMax op
maxpool_op = DownsampleFactorMax(maxpoolshp,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论