Fix crash in Alloc grad

theano/tensor/basic.py

@@ -2523,7 +2523,7 @@ class Alloc(gof.Op):
             new_order = list(x.broadcastable)
             idx = 0
             for i in range(x.ndim):
-                if not new_order[i]:
+                if i not in axis_broadcasted:
                     new_order[i] = idx
                     idx += 1
                 else:

theano/tensor/tests/test_basic.py

@@ -2062,6 +2062,19 @@ Allocb4GradTester = makeBroadcastTester(
 )
 
 
+# Partial un broadcast of a dimshuffled input
+AllocDimshuffleGradTester = makeBroadcastTester(
+    name='Allocb4GradTester',
+    op=lambda x: alloc(x.dimshuffle('x', 'x', 0), 1, s2, s3),
+    expected=(lambda x: numpy.zeros((1, s2, s3), dtype=x.dtype) + x),
+    grad=dict(
+        x1=(rand(s3),),
+        x2=(rand(s3),),
+        x3=(rand(s3),),
+    ),
+)
+
+
 class ApplyDefaultTestOp(theano.Op):
     def __init__(self, id):
         self.default_output = id