Fix the ability to specify a number of dimension as first argument of a random

theano/tensor/raw_random.py

@@ -210,11 +210,12 @@ def random_function(fn, dtype, *rfargs, **rfkwargs):
     - constants
     """
     @constructor
-    def f(ndim, *args, **kwargs):
+    def f(r, ndim, *args, **kwargs):
+        print 'f(', ndim, args, ')'
         if isinstance(ndim, int):
-            r, shape, args = args[0], args[1], args[2:]
+            shape, args = args[0], args[1:]
         else:
-            r, shape, args = ndim, args[0], args[1:]
+            shape = ndim
             if shape == () or shape == []:
                 shape = tensor.TensorConstant(type = tensor.lvector, data = shape)
             else: