def test_some_initialized(self):
with self.test_session() as sess:
x = tf.Variable(tf.zeros([]))
self.assertEqual([x], tdc._init_uninitialized(sess))
self.assertEqual(0, sess.run(x))
y = tf.assign_add(x, 1)
self.assertEqual([], tdc._init_uninitialized(sess))
self.assertEqual(1, sess.run(y))
self.assertEqual([], tdc._init_uninitialized(sess))
# If we had done initialize_all_variables we'd see 1.
self.assertEqual(2, sess.run(y))
评论列表
文章目录