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 5667b18 commit 4fa7994Copy full SHA for 4fa7994
pytensor/xtensor/rewriting/indexing.py
@@ -137,7 +137,7 @@ def lower_index(fgraph, node):
137
if basic_idx_axis and _non_consecutive_adv_indexing(aligned_idxs):
138
# Numpy moves advanced indexing dimensions to the front when they are not consecutive
139
# We need to transpose them back to the expected output order
140
- x_tensor_indexed_basic_dims = [out_dims[idx] for idx in basic_idx_axis]
+ x_tensor_indexed_basic_dims = [out_dims[axis] for axis in basic_idx_axis]
141
x_tensor_indexed_dims = [
142
dim for dim in out_dims if dim not in x_tensor_indexed_basic_dims
143
] + x_tensor_indexed_basic_dims
0 commit comments