torchviz模块的安装

可以使用以下命令安装 `torchviz`：

pip install torchviz

如果您使用的是 Anaconda，则可以使用以下命令安装：

conda install -c conda-forge torchviz

安装完成后，在代码中导入 `torchviz` 模块即可使用。例如：

from torchviz import make_dot

# 使用 make_dot 函数绘制计算图
dot = make_dot(outputs, params=dict(model.named_parameters()))
dot.render(filename='model', format='pdf')