pytorch自带pytorchviz

是的，PyTorch自带了一个名为pytorchviz的可视化工具。它可以帮助你可视化神经网络的计算图，以及方便地理解和调试模型。你可以使用pytorchviz将PyTorch模型的计算图导出为图像或PDF文件，并可选择是否显示变量名、形状和尺寸等信息。要使用pytorchviz，你需要先安装graphviz库，并且将其与PyTorch搭配使用。安装完依赖后，你可以通过导入torchviz来使用pytorchviz。