tf.add_to_collection()将tensor对象放入同一个集合
参数:
tf.add_to_collection(
name, //集合名
value
)
使用案例:
import tensorflow as tf
x = tf.constant(2)
y = tf.constant(3)
tf.add_to_collection("hehe", x)
tf.add_to_collection("hehe", y)
sess = tf.Session()
sess.run(tf.initialize_all_variables())
print(sess.run(tf.get_collection("hehe")))//[2, 3]