不同於keras的視覺化，keras.utils.plot_model。pytorch模型視覺化比較複雜，此處使用pytorchviz。

操作步驟

1 . 命令列安裝pytorchviz（之前要安裝graphviz，這個我keras時已經安裝）

pip install git+https://github.com/szagoruyko/pytorchviz

2 . 使用如下程式碼，儲存在統計目錄下的pdf檔案中，並開啟。

print(model)
g=make_dot(model(torch.rand(1,3,32,32).cuda()),params=dict(model.named_parameters()))
g.view()
 

其中，torch.rand(1,3,32,32).cuda()為偽造的資料，.cuda因為我的模型在gpu上。

效果