Skip to content

Commit ddcd988

Browse files
committed
Use DimShuffle instead of Reshape in ix_
1 parent 8e1cd56 commit ddcd988

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/tensor/basic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4551,7 +4551,7 @@ def ix_(*args):
45514551
new = as_tensor(new)
45524552
if new.ndim != 1:
45534553
raise ValueError("Cross index must be 1 dimensional")
4554-
new = new.reshape((1,) * k + (new.size,) + (1,) * (nd - k - 1))
4554+
new = new.dimshuffle(*(("x",) * k), 0, *(("x",) * (nd - k - 1)))
45554555
out.append(new)
45564556
return tuple(out)
45574557

0 commit comments

Comments
 (0)