提交 ad3b4ac7 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Don't return useless subtensor for `local_useless_slice`

上级 610d6199
...@@ -351,7 +351,8 @@ def local_useless_slice(fgraph, node): ...@@ -351,7 +351,8 @@ def local_useless_slice(fgraph, node):
new_idxs[dim] = slice(start, stop, step) new_idxs[dim] = slice(start, stop, step)
if change_flag or ((last_useful_idx + 1) < len(idxs)): if change_flag or ((last_useful_idx + 1) < len(idxs)):
out = x[tuple(new_idxs[: last_useful_idx + 1])] new_idxs = tuple(new_idxs[: last_useful_idx + 1])
out = x[new_idxs] if new_idxs else x
# Copy over previous output stacktrace # Copy over previous output stacktrace
copy_stack_trace(node.outputs, out) copy_stack_trace(node.outputs, out)
return [out] return [out]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论