Merge pull request #387 from lamblin/alloc_of_broadcasted

theano/tensor/basic.py

@@ -2542,7 +2542,7 @@ class Alloc(gof.Op):
 
     def c_code(self, node, name, inp, out, sub):
         # TODO: use the elemwise code generator here
-        if node.inputs[0].ndim == 0:
+        if python_all(node.inputs[0].broadcastable):
             # filling with a scalar is a common use of alloc
             # that we can implement relatively easily
             vv = inp[0]

theano/tensor/tests/test_basic.py

@@ -1208,6 +1208,8 @@ AllocTester = makeBroadcastTester(
         op = alloc,
         expected = (lambda x, *shp: numpy.zeros(shp, dtype=x.dtype) + x),
         good = dict(
+            correct01 = (rand(), numpy.int32(7)),
+            correct01_bcast = (rand(1), numpy.int32(7)),
             correct02 = (rand(), numpy.int32(4), numpy.int32(7)),
             correct12 = (rand(7), numpy.int32(4), numpy.int32(7)),
             correct13 = (rand(7), numpy.int32(2), numpy.int32(4), numpy.int32(7)),