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

Fix test values shape.

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