提交 a4425b96 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix ConstructSparseFromList.infer_shape

Now, the first argument contains the shape itself, so we should use its value, not its shape.
上级 b4889a2b
......@@ -3374,8 +3374,8 @@ class ConstructSparseFromList(gof.Op):
dtype=values.dtype)
def infer_shape(self, node, ishapes):
x, y, ilist = ishapes
return [x]
x = node.inputs[0]
return [[x[0], x[1]]]
def R_op(self, inputs, eval_points):
if None in eval_points[:2]:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论