change shape dimensions int32 to int64

theano/tensor/raw_random.py

@@ -184,7 +184,7 @@ class RandomFunction(gof.Op):
         """
         shape_ = tensor.as_tensor_variable(shape, ndim=1)
         if shape == ():
-            shape = shape_.astype('int32')
+            shape = shape_.astype('int64')
         else:
             shape = shape_
         assert shape.type.ndim == 1
@@ -363,7 +363,7 @@ def _infer_ndim_bcast(ndim, shape, *args):
         # post-condition: shape may still contain both symbolic and
         # non-symbolic things
         if len(pre_v_shape) == 0:
-            v_shape = tensor.constant([], dtype='int32')
+            v_shape = tensor.constant([], dtype='int64')
         else:
             v_shape = tensor.stack(pre_v_shape)
 
@@ -402,7 +402,7 @@ def _infer_ndim_bcast(ndim, shape, *args):
             (ndim, args_ndim), args)
 
     assert ndim == len(bcast)
-    return ndim, tensor.cast(v_shape, 'int32'), tuple(bcast)
+    return ndim, tensor.cast(v_shape, 'int64'), tuple(bcast)
 
 
 def _generate_broadcasting_indices(out_shape, *shapes):
@@ -520,13 +520,13 @@ def binomial(random_state, size=None, n=1, p=0.5, ndim=None,
     n = tensor.as_tensor_variable(n)
     p = tensor.as_tensor_variable(p)
     ndim, size, bcast = _infer_ndim_bcast(ndim, size, n, p)
-    if n.dtype == 'int64':
+    #if n.dtype == 'int64':
     # THIS WORKS AROUND A NUMPY BUG on 32bit machine
     # Erase when the following works on a 32bit machine:
     # numpy.random.binomial(
     #          n=numpy.asarray([2,3,4], dtype='int64'),
     #          p=numpy.asarray([.1, .2, .3], dtype='float64'))
-        n = tensor.cast(n, 'int32')
+    n = tensor.cast(n, 'int64')
     op = RandomFunction('binomial',
                         tensor.TensorType(dtype=dtype,
                                           broadcastable=(False,) * ndim))