如何对使用Pytork开发的CNN模型进行类似3d的可视化？

Question

目前，我有一个CNN模型，是我在Pytorch开发的。我使用hiddenlayer包创建了一个如图1所示的图像。但是，我想创建一个模型的图像，它应该看起来像图2中的东西。是否有任何包，我可以用来实现这一点

图1：

图2：

Answer 1

我个人使用figma.com并亲自“绘制”它，但是如果您想自动创建它，您应该查看thisgithub存储库，您可能会找到一个不错的工具