提交 d4ada676 authored 作者: Frederic's avatar Frederic

fix test, we must use the same paramete to both version!

上级 f44963cb
...@@ -155,7 +155,7 @@ class TestFourier(utt.InferShapeTester): ...@@ -155,7 +155,7 @@ class TestFourier(utt.InferShapeTester):
a = tensor.dmatrix() a = tensor.dmatrix()
f = function([a], self.op(a, n=10, axis=0)) f = function([a], self.op(a, n=10, axis=0))
a = numpy.random.rand(8, 6) a = numpy.random.rand(8, 6)
assert numpy.allclose(f(a), numpy.fft.fft(a)) assert numpy.allclose(f(a), numpy.fft.fft(a, 10, 0))
def test_infer_shape(self): def test_infer_shape(self):
a = tensor.dvector() a = tensor.dvector()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论