Network的写法一般有以下几种.

一、都写入__init__中

  像bert的网络模型就是这种,   好奇计算图是怎么工作的。一次init就能一直计算？

二、__init__和infer分离

  __init__中只存放网络层，infer中存放逻辑推理部分。

  好处是可以保证多次infer, 网络只初始化一次, 比如在对抗训练中。