提交 371c37b8 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix bug in extract_constant

上级 744a069e
...@@ -2834,7 +2834,7 @@ def extract_constant(x): ...@@ -2834,7 +2834,7 @@ def extract_constant(x):
if x.owner and isinstance(x.owner.op, ScalarFromTensor): if x.owner and isinstance(x.owner.op, ScalarFromTensor):
x = x.owner.inputs[0] x = x.owner.inputs[0]
else: else:
x = tensor.tensor_from_scalar(x) x = tensor_from_scalar(x)
return x return x
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论