提交 32fff4aa authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add test that mean() over float16 doesn't overflow for large arrays.

上级 78aa6276
...@@ -4596,6 +4596,12 @@ class T_mean(unittest.TestCase): ...@@ -4596,6 +4596,12 @@ class T_mean(unittest.TestCase):
except AttributeError: except AttributeError:
self.fail() self.fail()
def test_mean_f16(self):
x = tensor.vector(dtype='float16')
y = x.mean()
f = theano.function([x], y)
utt.assert_allclose(f(numpy.ones((100000,), dtype='float16')), 1.0)
def test0(self): def test0(self):
# Simple test... # Simple test...
x = tensor.vector() x = tensor.vector()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论