====== 2-图形化model-查看模型的基本结构 ======

===== torch的module的可视化  =====
<code>
from torchview import draw_graph
model_graph = draw_graph(model, input_size=(1, 1, 28, 28), save_graph=True, expand_nested=True)
</code>