1. 程式人生 > >TensorFlow基礎用法——tf.control_dependencies()函式用法

TensorFlow基礎用法——tf.control_dependencies()函式用法

Ingredient:

tf.control_dependencies()函式是用來控制計算流圖的,也就是給圖中的某些計算指定順序。有的時候我們想要指定某些操作執行的依賴關係,比如想要讓引數先更新,然後再獲取引數更新後的值等。

tf.control_dependencies(control_inputs)會返回一個控制依賴的上下文管理器,使用了with關鍵字就可以讓在這個上下文環境中的操作都在control_inputs 執行,比如:

with g.control_dependencies([a, b, c]):
  # `d` and `e` will only run after `a`, `b`, and `c` have executed.
  d = ...
  e = ...

d、e的操作會在a、b、c的操作執行完之後再執行。

參考: