醉酒诗人的博客

方法一： 自己下载graphviz程序，然后编写make_dot函数，然后进行调用。 下载graphviz程序可以参考：文章一、文章二 参考的demo如下： import torch from torch.autograd import Variable import torch.nn as nn from graphviz import Digraph class CNN(nn.Module): def __init__(self): super(CNN, self).__in