阅读(14.7k) 书签 (0)

TensorFlow的graph_util模块

2018-10-17 17:25 更新

模块: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 的输入字符串获取形状的便利函数.