如何对使用pytorch开发的CNN模型进行3-d可视化？

Question

目前，我有一个在Pytorch中开发的CNN模型。我已经使用hiddenlayer程序包来创建如图1所示的图像。但是，我想创建一个模型的图像，该图像看起来应该类似于图2中的内容。是否可以使用任何程序包来实现此目的？

图片1：

图片2：

Answer 1

我个人使用figma.com自己“绘制”它，但是如果要自动创建它，则应签出this github存储库，您可能会找到一个不错的工具。