提交 07fe078c authored 作者: Li's avatar Li

get rid of warnings when testing

上级 b9ee8cfa
...@@ -367,12 +367,12 @@ class T_Images2Neibs(unittest_tools.InferShapeTester): ...@@ -367,12 +367,12 @@ class T_Images2Neibs(unittest_tools.InferShapeTester):
f() f()
def test_infer_shape(self): def test_infer_shape(self):
shape = (100, 40, 6, 6) shape = (100, 40, 6, 3)
images = numpy.ones(shape).astype('float32') images = numpy.ones(shape).astype('float32')
x = T.ftensor4() x = T.ftensor4()
f = self._compile_and_check([x], f = self._compile_and_check([x],
[images2neibs( [images2neibs(
x, neib_shape=(2,2), x, neib_shape=(2,1),
mode='valid')], mode='valid')],
[images], [images],
Images2Neibs Images2Neibs
...@@ -384,17 +384,17 @@ class T_Images2Neibs(unittest_tools.InferShapeTester): ...@@ -384,17 +384,17 @@ class T_Images2Neibs(unittest_tools.InferShapeTester):
[images], [images],
Images2Neibs Images2Neibs
) )
shape = (100, 40, 5, 5) shape = (100, 40, 5, 4)
images = numpy.ones(shape).astype('float32') images = numpy.ones(shape).astype('float32')
x = T.ftensor4() x = T.ftensor4()
f = self._compile_and_check([x], f = self._compile_and_check([x],
[images2neibs( [images2neibs(
x, neib_shape=(2,2), x, neib_shape=(2,1),
mode='ignore_borders')], mode='ignore_borders')],
[images], [images],
Images2Neibs Images2Neibs
) )
shape = (100, 40, 5, 5) shape = (100, 40, 5, 3)
images = numpy.ones(shape).astype('float32') images = numpy.ones(shape).astype('float32')
x = T.ftensor4() x = T.ftensor4()
f = self._compile_and_check([x], f = self._compile_and_check([x],
...@@ -405,7 +405,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester): ...@@ -405,7 +405,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester):
Images2Neibs Images2Neibs
) )
shape = (100, 40, 6, 6) shape = (100, 40, 6, 7)
images = numpy.ones(shape).astype('float32') images = numpy.ones(shape).astype('float32')
x = T.ftensor4() x = T.ftensor4()
f = self._compile_and_check([x], f = self._compile_and_check([x],
...@@ -415,7 +415,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester): ...@@ -415,7 +415,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester):
[images], [images],
Images2Neibs Images2Neibs
) )
shape = (100, 40, 5, 5) shape = (100, 40, 5, 10)
images = numpy.ones(shape).astype('float32') images = numpy.ones(shape).astype('float32')
x = T.ftensor4() x = T.ftensor4()
f = self._compile_and_check([x], f = self._compile_and_check([x],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论