graph_test.py 827 B

1234567891011121314151617181920212223242526272829303132
  1. import tensorflow as tf
  2. g1 = tf.Graph()
  3. with g1.as_default():
  4. v = tf.get_variable(
  5. "v", initializer=tf.zeros_initializer()(shape=[1])
  6. )
  7. g2 = tf.Graph()
  8. with g2.as_default():
  9. v = tf.get_variable(
  10. "v", initializer=tf.ones_initializer()(shape=[1])
  11. )
  12. with tf.Session(graph=g1) as sess:
  13. tf.global_variables_initializer().run()
  14. with tf.variable_scope("", reuse=True):
  15. print(sess.run(tf.get_variable("v")))
  16. with tf.Session(graph=g2) as sess:
  17. tf.global_variables_initializer().run()
  18. with tf.variable_scope("", reuse=True):
  19. print(sess.run(tf.get_variable("v")))
  20. # gpu acceleration
  21. a = tf.constant([1.0, 3.0], name="a")
  22. b = tf.constant([3.0, 6.0], name="b")
  23. g = tf.Graph()
  24. with g.device('/gpu:0'):
  25. result = a+b
  26. sess = tf.Session()
  27. print (sess.run(result))