如何将MXNet图形可视化存储为文件中的图像？

Question

根据this页面，可以使用mx.viz.plot_network(net)显示MXNet计算图。

但这只适用于Jupyter笔记本。没有Jupyter笔记本，我该怎么做？是否可以将可视化文件保存为文件中的图像？

Answer 1

mx.viz.plot_network返回一个graphviz.dot.Digraph对象，它可以像任何其他Digraph对象一样呈现为文件。

以下是一个例子：

# Store the Digraph in a variable
graph = mx.viz.plot_network(net)

# Pick an image format from this list: http://www.graphviz.org/doc/info/output.html
graph.format = 'png'

# Choose a filename and render the image.
graph.render('graph')

以上代码将在＆＃39; graph.png＆＃39;中呈现图表。在当前目录中。