TensorFlow的graph_util模块

由 Carrie 创建, 最后一次修改 2017-10-17

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

以上内容是否对您有帮助:
二维码
建议反馈
二维码