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

tf.graph_util.extract_sub_graph

extract_sub_graph(
    graph_def,
    dest_nodes
)

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

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

參數(shù)：

• graph_def：一個(gè) graph_pb2.GraphDef 原型.
• dest_nodes：指定目標(biāo)節(jié)點(diǎn)名稱(chēng)的字符串列表.

返回值：

該操作返回子圖的 GraphDef.

錯(cuò)誤類(lèi)型引發(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上運(yùn)行,則返回真,否則返回假.

參數(shù)：

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

返回值：

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