TensorFlow将计算图导出为XML,JSON等

发布时间：2020-12-16 07:57:42 所属栏目：百科来源：网络整理

导读：我想将TensorFlow计算图导出为XML或类似的东西,以便我可以使用外部程序对其进行修改,然后重新导入它.我找到了 Meta Graph,但这是以二进制格式导出的,我不知道如何修改. 这种能力是否存在？ TensorFlow的数据流图的本机序列化格式使用 protocol buffers,它具

我想将TensorFlow计算图导出为XML或类似的东西,以便我可以使用外部程序对其进行修改,然后重新导入它.我找到了 Meta Graph,但这是以二进制格式导出的,我不知道如何修改.

这种能力是否存在？

TensorFlow的数据流图的本机序列化格式使用 protocol buffers,它具有许多不同语言的绑定.您可以生成应该能够解析来自两个消息模式的二进制数据的代码： tensorflow.GraphDef(较低级别的表示)和 tensorflow.MetaGraphDef(较高级别的表示,其中包括GraphDef以及有关如何解释某些图中的节点).

如果目标语言没有协议缓冲区实现,则可以从Python协议缓冲区对象生成JSON.例如,以下代码生成一个包含GraphDef的JSON表示的字符串：

import tensorflow as tf
from google.protobuf import json_format

with tf.Graph().as_default() as graph:
  # Add nodes to the graph...

graph_def = graph.as_graph_def()

json_string = json_format.MessageToJson(graph_def)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!