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 f000bbb commit 2de9566Copy full SHA for 2de9566
pytensor/xtensor/rewriting/shape.py
@@ -123,7 +123,7 @@ def lower_transpose(fgraph, node):
123
@node_rewriter([Squeeze])
124
def local_squeeze_reshape(fgraph, node):
125
"""Rewrite Squeeze to tensor.squeeze."""
126
- x = node.inputs[0]
+ [x] = node.inputs
127
x_tensor = tensor_from_xtensor(x)
128
x_dims = x.type.dims
129
dims_to_remove = node.op.dims
0 commit comments