module=nn.ModuleList([nn.Linear(5, 10])
module.extend([nn.Linear(10, 5])
如何使用TensorFlow2达到同样的效果?
如何使用TensorFlow2实现pytorch中的nn.ModuleList
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-
module=nn.ModuleList([nn.Linear(5, 10])
module.extend([nn.Linear(10, 5])
如何使用TensorFlow2达到同样的效果?