标签： python keras pytorch
Keras具有很好的功能，可以将模型架构另存为框图
keras.utils.plot_model(self.model, plot_filepath.as_posix(), show_shapes=True)
这可以在PyTorch中实现吗？
This answer讨论了与Keras类似的打印摘要。是否有类似的东西可以保存模型架构的框图？