【TensorFlow动手玩】常用集合: Variable, Summary, 自定义

本文介绍了TensorFlow中常用集合的使用,包括Variable和Summary。Variable用于存储和更新模型参数,创建后需要初始化,并可从集合中获取。Summary用于监测网络状态,可以通过tf.scalar_summary添加并用TensorBoard查看。同时,文章讨论了如何自定义集合,特别是在创建损失函数时的操作。
摘要由CSDN通过智能技术生成

集合

tensorflow用集合colletion组织不同类别的对象。tf.GraphKeys中包含了所有默认集合的名称。

collection提供了一种“零存整取”的思路:在任意位置,任意层次都可以创造对象,存入相应collection中;创造完成后,统一从一个collection中取出一类变量,施加相应操作。

例如,tf.Optimizer只优化tf.GraphKeys.TRAINABLE_VARIABLES中的变量。

本文介绍几个常用集合
- Variable集合:模型参数
- Summary集合:监测
- 自定义集合

Variable

Variable被收集在名为tf.GraphKeys.VARIABLEScolletion

定义

Tensorflow使用Variable类表达、更新、存储模型参数。

Variable是在可变更的,具有保持性的内存句柄,存储着Tensor。必须使用Tensor进行初始化。

k = tf.Variable(tf.random_normal([]), name='k')

创建的Variable被添加到默认的collection中。

初始化

在整个session运行之前,图中的全部Variable必须被初始化。

sess = tf.Session()
init = tf.initialize_all_variables() 
sess.run(init)

在执行完初始化之后,Variable中的值生成完毕,不会再变化。

特别强调Variable的值在sess.run(init)之后就确

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值