如何操作TensorFlow张量图的节点

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

tf.graph_util.extract_sub_graph

extract_sub_graph(
    graph_def,
    dest_nodes
)

定义在:tensorflow/python/framework/graph_util_impl.py

提取可以到达 “dest_nodes” 中的任何节点的子图.

参数:

  • graph_def:一个 graph_pb2.GraphDef 原型.
  • dest_nodes:指定目标节点名称的字符串列表.

返回值:

该操作返回子图的 GraphDef.

错误类型引发:

  • 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上运行,则返回真,否则返回假.

参数:

  • node:要分配要给设备的节点;可以是 ops.Operation 也可以是 NodeDef.
  • pin_variables_on_cpu:当为真时,如果 node_def 表示与变量相关的操作,则此函数将返回假.

返回值:

如果给定的节点必须在 CPU上运行,则返回真,否则为返回假.

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