tensor.opt - added dimshuffle_as_view optimization

theano/tensor/opt.py

@@ -203,6 +203,15 @@ def local_dimshuffle_lift(node):
         else:
             return DimShuffle(iinput.type.broadcastable, new_order, inplace).make_node(iinput).outputs
 
+@register_specialize
+@gof.local_optimizer([])
+def dimshuffle_as_view(node):
+    op = node.op
+    if not isinstance(op, DimShuffle) or op.inplace:
+        return False
+    new_op = DimShuffle(op.input_broadcastable, op.new_order, inplace=True)
+    return [new_op(*node.inputs)]
+
 
 register_canonicalize(local_dimshuffle_lift)
 register_specialize(local_dimshuffle_lift)