今天小编就为大家分享一篇画pytorch模型图,以及参数计算的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

刚入pytorch的坑，代码还没看太懂。之前用keras用习惯了，第一次使用pytorch还有些不适应，希望广大老司机多多指教。 首先说说，我们如何可视化模型。在keras中就一句话，keras.summary()，或者plot_model()，就可以把模型展现的淋漓尽致。 但是pytorch中好像没有这样一个api让我们直观的看到模型的样子。但是有网友提供了一段代码，可以把模型画出来，对我来说简直就是如有神助啊。 话不多说，上代码吧。 import torch from torch.auto