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

Question

根据this页面，MXNet计算图可以使用mx.viz.plot_network(net)进行可视化。

但这仅适用于 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')

以上代码将在当前目录中的“graph.png”中呈现图形。