TensorFlow学习－－tf.add_to_collection与tf.get_collection使用

tf.add_to_collection–向当前计算图中添加张量集合

tf.get_collection–返回当前计算图中手动添加的张量集合

#!/usr/bin/python# coding:utf-8import tensorflow as tfv1 = tf.get_variable('v1', shape=[3], initializer=tf.ones_initializer())v2 = tf.get_variable('v2', shape=[5], initializer=tf.random_uniform_initializer(maxval=-1., minval=1., seed=0))# 向当前计算图中添加张量集合tf.add_to_collection('v', v1)tf.add_to_collection('v', v2)with tf.Session() as sess:    sess.run(tf.global_variables_initializer())    # 返回当前计算图中手动添加的张量集合    v = tf.get_collection('v')    print v    print v[0].eval()    print v[1].eval()

输出：

[<tf.Variable 'v1:0' shape=(3,) dtype=float32_ref>, <tf.Variable 'v2:0' shape=(5,) dtype=float32_ref>][ 1.  1.  1.][ 0.79827476 -0.9403336  -0.69752836  0.90343738  0.90295386]