Solved by jpeg729 in post #2 
              
              
                When using torch.nn.Linear for example, the nn.Linear class looks after initialising and using the parameters that it requires. 
When using the torch.nn.Functional.linear variant, it is up to you to provide the parameters on each forward pass. 
Basically instead of 
class Model(nn.Module):
    def _…