一、环境准备
在搭建生成对抗网络时,我发现会用到一个将张量从计算图剥离的函数:detach(),这进一步激发了我的求知欲,我想知道这个detach函数到底是怎么执行的,我猜想把计算图可视化应该可以解决这个疑问,通过万能的互联网果真是有此类工具PyTorchViz,Github地址为:https://github.com/szagoruyko/pytorchviz,根据人家给出的Readme文件可知在进行代码编写前首先需要进行环境配置,安装人家需要的第三方库。
具体步骤如下:
1.在工作环境(比如通过Anconda创建的虚拟环境)下依次安装graphviz和torchviz:
1)安装graphviz
pip install graphviz
2)