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 2e22d34 commit 79d9662Copy full SHA for 79d9662
pytensor/tensor/extra_ops.py
@@ -2066,9 +2066,7 @@ def unpack(
2066
raise ValueError("Cannot unpack an empty list of shapes.")
2067
2068
n_splits = len(packed_shapes)
2069
- split_size = [
2070
- prod(shape, no_zeros_in_input=True).astype(int) for shape in packed_shapes
2071
- ]
+ split_size = [prod(shape).astype(int) for shape in packed_shapes]
2072
unpacked_tensors = split(flat_tensor, splits_size=split_size, n_splits=n_splits)
2073
2074
return tuple(
0 commit comments