提交 86c8a2ec authored 作者: Ian Goodfellow's avatar Ian Goodfellow

Fred modified an optimization to disable itself in a case where it has a

bug
上级 fcb0bc9e
...@@ -1911,6 +1911,8 @@ def local_subtensor_of_alloc(node): ...@@ -1911,6 +1911,8 @@ def local_subtensor_of_alloc(node):
nw_val = val[tuple(val_slices)] nw_val = val[tuple(val_slices)]
nw_dims += dims[len(slices):] nw_dims += dims[len(slices):]
if nw_val.ndim > len(nw_dims):
return False
rval = T.alloc(nw_val, *nw_dims) rval = T.alloc(nw_val, *nw_dims)
if type(rval) not in (list, tuple): if type(rval) not in (list, tuple):
rval = [rval] rval = [rval]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论