A clean approach would be to define the desired parameters once (via nn.Parameter in the __init__ method of the model) and apply them through the functional API in the forward method (e.g. via F.linear).