TensorFlow的graph_util模塊

2018-10-17 17:25 更新

模塊:tf.graph_util

定義在:tensorflow/python/framework/graph_util.py

在 python 中操作張量圖的幫助器.

使用函數(shù)

convert_variables_to_constants(...):用相同值的常量替換圖表中的所有變量.

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

must_run_on_cpu(...):如果給定的 node_def 必須在 CPU 上運行,則返回 True,否則為 False.

remove_training_nodes(...):調(diào)整不需要推理的節(jié)點.

tensor_shape_from_node_def_name(...):從 NodeDef 的輸入字符串獲取形狀的便利函數(shù).

以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號