提交 8cda5a86 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix test values shape.

上级 59e08c0d
...@@ -39,7 +39,7 @@ class OpDecoratorTests(utt.InferShapeTester): ...@@ -39,7 +39,7 @@ class OpDecoratorTests(utt.InferShapeTester):
x = dmatrix('x') x = dmatrix('x')
x.tag.test_value = np.zeros((2, 2)) x.tag.test_value = np.zeros((2, 2))
y = dvector('y') y = dvector('y')
y.tag.test_value = [0, 0] y.tag.test_value = [0, 0, 0, 0]
@as_op([dmatrix, dvector], dvector) @as_op([dmatrix, dvector], dvector)
def cumprod_plus(x, y): def cumprod_plus(x, y):
...@@ -55,7 +55,7 @@ class OpDecoratorTests(utt.InferShapeTester): ...@@ -55,7 +55,7 @@ class OpDecoratorTests(utt.InferShapeTester):
x = dmatrix('x') x = dmatrix('x')
x.tag.test_value = np.zeros((2, 2)) x.tag.test_value = np.zeros((2, 2))
y = dvector('y') y = dvector('y')
y.tag.test_value = [0, 0] y.tag.test_value = [0, 0, 0, 0]
def infer_shape(node, shapes): def infer_shape(node, shapes):
x, y = shapes x, y = shapes
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论