Solved by ptrblck in post #2 
              
              
                Both models define valid forward methods (besides the missing return statement) and it depends on your use case, which one is correct. 
If you would like to apply the same parameters for both inputs (weight sharing), the first definition is correct, while the second model will use two different laye…