We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
local_useless_slice
1 parent 2266bd8 commit 14c4eaaCopy full SHA for 14c4eaa
pytensor/tensor/rewriting/subtensor.py
@@ -351,7 +351,8 @@ def local_useless_slice(fgraph, node):
351
new_idxs[dim] = slice(start, stop, step)
352
353
if change_flag or ((last_useful_idx + 1) < len(idxs)):
354
- out = x[tuple(new_idxs[: last_useful_idx + 1])]
+ new_idxs = tuple(new_idxs[: last_useful_idx + 1])
355
+ out = x[new_idxs] if new_idxs else x
356
# Copy over previous output stacktrace
357
copy_stack_trace(node.outputs, out)
358
return [out]
0 commit comments