安装 torchviz： pip install torchviz 示例代码：下面是一个简单的示例代码，展示如何使用 torchviz 可视化计算图： pythonimport torchfrom torchviz import make_dot# 创建一个简单的模型model = torch.nn.Sequential(torch.nn.Linear(2, 2),torch.nn.