Added back the local_dimshuffle_lift opt back

theano/tensor/elemwise.py

@@ -133,7 +133,7 @@ class DimShuffle(Op):
 
     _f16_ok = True
     check_input = False
-    __props__ = ("input_broadcastable", "new_order")
+    __props__ = ("input_broadcastable", "new_order", "inplace")
 
     def __init__(self, input_broadcastable, new_order, inplace=True):
         input_broadcastable = tuple(input_broadcastable)

theano/tensor/opt.py

@@ -649,11 +649,14 @@ def local_lift_transpose_through_dot(node):
         copy_stack_trace(node.inputs[0], ret)
         return ret
 
+register_canonicalize(local_dimshuffle_lift)
+register_specialize(local_dimshuffle_lift)
 
 ######################
 # Casting operations #
 ######################
 
+
 @register_canonicalize
 @register_specialize
 @gof.local_optimizer([T.TensorFromScalar])