How to use TensorFlow2 to implement nn.ModuleList in pytorch

userpEsl1 0
Viewed 17

module=nn.ModuleList([nn.Linear(5, 10]) module.extend([nn.Linear(10, 5]) How to use TensorFlow2 to achieve the same effect?

tensorflow2.0 pytorch
0 Answers
Related