Python中graphviz库的简单用法

import VisualizeGraph as viz

x = Variable(torch.randn(1, 3, args.inWidth, args.inHeight))    # 产生随机X

if args.onGPU == True:
            x = x.cuda()

y = model.forward(x)                #传入网络模型
g = viz.make_dot(y)                 #可视化有向图g
g.render(args.savedir + '/model.png', view=False)    #保存有向图 view =False代表运行时不弹出图片