如何操作TensorFlow張量圖的節(jié)點

tf.graph_util.extract_sub_graph

extract_sub_graph(
    graph_def,
    dest_nodes
)

定義在：tensorflow/python/framework/graph_util_impl.py

提取可以到達 “dest_nodes” 中的任何節(jié)點的子圖.

參數(shù)：

• graph_def：一個 graph_pb2.GraphDef 原型.
• dest_nodes：指定目標節(jié)點名稱的字符串列表.

返回值：

該操作返回子圖的 GraphDef.

錯誤類型引發(fā)：

• TypeError：如果 'graph_def' 不是 graph_pb2.GraphDef 原型.

tf.graph_util.must_run_on_cpu

must_run_on_cpu(
    node,
    pin_variables_on_cpu=False
)

定義在：tensorflow/python/framework/graph_util_impl.py

如果給定的 node_def 必須在CPU上運行,則返回真,否則返回假.

參數(shù)：

• node：要分配要給設(shè)備的節(jié)點；可以是 ops.Operation 也可以是 NodeDef.
• pin_variables_on_cpu：當為真時,如果 node_def 表示與變量相關(guān)的操作,則此函數(shù)將返回假.

返回值：

如果給定的節(jié)點必須在 CPU上運行,則返回真,否則為返回假.