提交 371e781f authored 作者: James Bergstra's avatar James Bergstra

Bugfix to get_constant_value()

上级 2153cabf
......@@ -69,8 +69,7 @@ def get_constant_value(v):
if v.owner and v.owner.op == T.fill:
shape, val = v.owner.inputs
# fill(a,b) fills the shape of 'a' filled with 'b'
rval, rshapes = get_constant_value(val)
return rval, rshapes + [shape]
return get_constant_value(val)
raise TypeError(v)
def scalarconsts_rest(inputs):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论