Skip to content

Commit ff63514

Browse files
committed
Check shape length matches dims in XTensorType
1 parent 28a4d86 commit ff63514

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

pytensor/xtensor/type.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,10 @@ def __init__(
4949
self.shape = (None,) * len(self.dims)
5050
else:
5151
self.shape = tuple(shape)
52+
if len(self.shape) != len(self.dims):
53+
raise ValueError(
54+
f"Shape {self.shape} must have the same length as dims {self.dims}"
55+
)
5256
self.ndim = len(self.dims)
5357
self.name = name
5458

0 commit comments

Comments
 (0)