Add shuffle_rows function to RandomStreams, inspired from numpy.random.shuffle

theano/tensor/randomstreams.py

@@ -6,7 +6,7 @@ import numpy
 
 from theano.compile import module, In, Component
 from theano.gof import Container
-from theano.tensor import raw_random
+from theano.tensor import raw_random, reorder_row_elements
 
 class RandomStreamsInstance(object):
     """RandomStreamsInstance"""
@@ -189,3 +189,10 @@ class RandomStreams(Component):
         """
         return self.gen(raw_random.multinomial, *args, **kwargs)
 
+    def shuffle_rows(self, input):
+        """Return a variable with every row (rightmost index) shuffled"""
+        perm = self.permutation(input.ndim-1, input.shape[:-1], input.shape[-1])
+        shuffled = reorder_row_elements(input, perm)
+        return shuffled
+
+