技术 #PyTorch 利用torch.fx提取PyTorch网络结构信息绘制网络结构图 2022/05/20 介绍torch.fx的基础信息。利用torch.fx将PyTorch网络转换为一个图结构，然后使用Graphviz可视化这个图