Solved by jpeg729 in post #3 
              
              
                A naive approach would indeed be to construct m using nn.ModuleList() and a for loop. 
But I think we parallelise the whole lot. The following idea should work, but I haven’t checked all the details. 
The input x is shared by all subnetworks, so a single Linear layer of size m*100 followed by separa…