Add rng_owner attribute to RandomStreams as well

theano/tensor/shared_randomstreams.py

@@ -5,6 +5,7 @@ __docformat__ = "restructuredtext en"
 
 import copy
 import numpy
+import weakref
 
 from theano.compile.sharedvalue import (SharedVariable, shared_constructor,
                                         shared)
@@ -132,6 +133,8 @@ class RandomStreams(raw_random.RandomStreamsBase):
         """
         seed = int(self.gen_seedgen.randint(2 ** 30))
         random_state_variable = shared(numpy.random.RandomState(seed))
+        # Add a reference to distinguish from other shared variables
+        random_state_variable.rng_owner = weakref.ref(self)
         new_r, out = op(random_state_variable, *args, **kwargs)
         out.rng = random_state_variable
         out.update = (random_state_variable, new_r)