首页 > 其他 > 详细

tf.control_dependencies

时间:2019-08-30 00:07:30      阅读:90      评论:0      收藏:0      [点我收藏+]

tf.control_dependencies()是用来控制计算流图的,给图中的某些节点指定计算的顺序。

原型:

tf.control_dependencies(self, control_inputs)

该函数接受的参数control_inputs,是Operation或者Tensor构成的list。

例子:确保获得更新后的参数:

opt = tf.train.Optimizer().minize(loss)

 

with tf.control_dependencies([opt]): #先执行opt

  updated_weight = tf.identity(weight)  #再执行该操作

 

with tf.Session() as sess:

  tf.global_variables_initializer().run()

  sess.run(updated_weight, feed_dict={...}) # 这样每次得到的都是更新后的weight

tf.control_dependencies

原文:https://www.cnblogs.com/HYWZ36/p/11432681.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!