提交 be93a84e authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix opt crash. fix gh-5316

上级 b9e43354
......@@ -133,7 +133,8 @@ def local_reshape_dimshuffle(node):
return False
else:
offset += 1
return [T.reshape(input_.owner.inputs[0], node.inputs[1])]
return [T.reshape(input_.owner.inputs[0], node.inputs[1],
ndim=node.outputs[0].ndim)]
return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论