Add drop property to Dimshuffle

Author: ricardoV94

pytensor/tensor/elemwise.py

@@ -173,6 +173,7 @@ def __init__(self, input_broadcastable, new_order):
         # List of dimensions of the output that are broadcastable and were not
         # in the original input
         self.augment = sorted([i for i, x in enumerate(new_order) if x == "x"])
+        self.drop = drop
 
         if self.inplace:
             self.view_map = {0: [0]}