Solved by alex.veuthey in post #4 
              
              
                Sure, but a few explanations first: 
nn.Module: very practical for learnable layers (will handle backward pass automatically once you define the forward() method), but it needs to __init__ first, and then will be called in the forward(x) method with some input tensor x. 
nn.functional: shortcut func…