Function to initialize a nn_module Forward functions works with a list. The entries of the list are the input of the subnetworks
Useful links