when shape or broadcasting specified do not set the default shape and broadcasting

Author: ferrine

pytensor/tensor/type.py

@@ -144,12 +144,14 @@ def __init__(
     def clone(
         self, dtype=None, shape=None, broadcastable=None, **kwargs
     ) -> "TensorType":
-        if broadcastable is None:
+        if broadcastable is None and shape is None:
+            # only default to passby broadcast when shape is also None
+            # otherwise shape changes affect broadcasting patterns and
+            # inconsistencies
             broadcastable = self.broadcastable
+            shape = self.shape
         if dtype is None:
             dtype = self.dtype
-        if shape is None:
-            shape = self.shape
         return type(self)(dtype, shape, name=self.name, broadcastable=broadcastable)
 
     def filter(self, data, strict=False, allow_downcast=None):