提交 ee70708b authored 作者: Sina Honari's avatar Sina Honari

adding the fix for Reshape.infer_shape

上级 c26f8cbf
...@@ -4417,7 +4417,7 @@ class Reshape(Op): ...@@ -4417,7 +4417,7 @@ class Reshape(Op):
if ele == -1: if ele == -1:
requ[i] = missing requ[i] = missing
elif crit == 1: # we reshape to -1 elif crit == 1: # we reshape to -1
requ = [mul(*ishapes[0])] requ = [mul(*ishapes[0])] if ishapes[0] else [1]
elif crit > 1: elif crit > 1:
raise ValueError('shape argument to Reshape.perform' raise ValueError('shape argument to Reshape.perform'
' must have at most one entry equal to -1') ' must have at most one entry equal to -1')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论