# -*- coding: utf-8 -*-
"""
Created on Thu Jun 05 17:48:31 2014
@author: Administrator
"""
import theano
import numpy as np
import theano.tensor as T
from theano.sandbox.rng_mrg import MRG_RandomStreams
from theano.tensor.shared_randomstreams import RandomStreams
class Graph():
def __init__(self, seed=123):
self.rng = RandomStreams(seed)
self.y = self.rng.uniform(size=(1,))
def copy_random_state(g1, g2):
if isinstance(g1.rng, MRG_RandomStreams):
g2.rng.rstate = g1.rng.rstate
for (su1, su2) in zip(g1.rng.state_updates, g2.rng.state_updates):
su2[0].set_value(su1[0].get_value())
if __name__ == "__main__":
g1 = Graph(seed=123)
f1 = theano.function([], g1.y)
g2 = Graph(seed=884)
f2 = theano.function([], g2.y)
print 'f1 return ', f1()
print 'f2 return ', f2()
print 'after copy random state'
copy_random_state(g1, g2)
print 'f1 return ', f1()
print 'f2 return ', f2()