pytorch tensorboard模型结构显示

转自：https://www.cnblogs.com/jfdwd/p/11445135.html

重点关注最后两句，其中使用了python的上下文管理，with 语句，可以避免因w.close未写造成的问题。推荐使用此方式。
使用 w.add_graph(model, (dummy_input,))，其中第一个参数为需要保存的模型，第二个参数为输入值，tuple类型。打开tensorboard控制台，可得到如下结果。