TensorFlow学习--tf.add_to_collection与tf.get_collection使用
2017-11-13 16:36
507 查看
tf.add_to_collection–向当前计算图中添加张量集合
tf.get_collection–返回当前计算图中手动添加的张量集合
输出:
tf.get_collection–返回当前计算图中手动添加的张量集合
#!/usr/bin/python # coding:utf-8 import tensorflow as tf v1 = 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]
相关文章推荐
- 如何利用tf.add_to_collection、tf.get_collection以及tf.add_n来简化正则项的计算
- TensorFlow学习--tf.get_variable使用
- tf.add_to_collection,tf.get_collection和tf.add_n的用法
- tensorflow: tf.add_to_collection 探究
- tf.add_to_collection、tf.get_collection、tf.add_n用法浅析
- tensorflow学习——tf.get_collection(), tf.identity()
- tf.add_to_collection()
- 简单Tensorflow线性拟合类及tf.get_variable()使用示例
- Tensorflow中tf.get_variable和tf.variable_scope的使用
- tensorflow将训练好的模型freeze,即将权重固化到图里面,并使用该模型进行预测(tf.graph_util.convert_variables_to_constants函数)
- 【tensorflow】scope的使用以及tf.Variable()和tf.get_variable()的区别
- TensorFlow 学习:tensor.get_shape() 与 tf.shape(a)
- TensorFlow 学习(一)—— tf.get_variable() vs tf.Variable(),tf.name_scope() vs tf.variable_scope()
- tensorflow中使用tf.variable_scope和tf.get_variable的ValueError
- tensorflow学习笔记--tf.get_variable、tf.Variable
- 【tensorflow 学习】tf.get_variable()和tf.Variable()的区别
- tensorflow:3.1)add_to_collection和L2正则化
- TensorFlow 学习(一)—— tf.get_variable() vs tf.Variable(),tf.name_scope() vs tf.variable_scope()
- ejb 学习:遇到错误:LazyInitializationException.<init> : failed to lazily initialize a collection of role: