阅读(2035)
赞(13)
TensorFlow的graph_util模块
2017-10-17 17:07:52 更新
模块:tf.graph_util
定义在:tensorflow/python/framework/graph_util.py
在 python 中操作张量图的帮助器.
使用函数
convert_variables_to_constants(...):用相同值的常量替换图表中的所有变量.
extract_sub_graph(...):提取可以到达“dest_nodes”中的任何节点的子图.
must_run_on_cpu(...):如果给定的 node_def 必须在 CPU 上运行,则返回 True,否则为 False.
remove_training_nodes(...):调整不需要推理的节点.
tensor_shape_from_node_def_name(...):从 NodeDef 的输入字符串获取形状的便利函数.