pytorch 可视化笔记2：可视化AlexNet

import torch
import torch.nn as nn
import torchviz
import torchvision

model = torchvision.models.AlexNet()
x = torch.randn(1,3,224,224)
y = model(x)

torchviz.make_dot(y, params=dict(model.named_parameters()))