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.
1 parent 79ff97a commit 335ca1eCopy full SHA for 335ca1e
pytensor/tensor/rewriting/subtensor.py
@@ -1,3 +1,4 @@
1
+import itertools
2
import sys
3
from collections.abc import Iterable
4
@@ -1700,7 +1701,7 @@ def local_join_subtensors(fgraph, node):
1700
1701
return
1702
1703
for subtensor1_idx, (subtensor1, subtensor2) in enumerate(
- zip(tensors[:-1], tensors[1:])
1704
+ itertools.pairwise(tensors)
1705
):
1706
# Check that two consecutive Subtensors are operating on the same base tensor
1707
if not (
0 commit comments