W1 will be updated as it’s wrapped in nn.Parameter, but w1 will not, as it’s a simple floating number.
Also, the initialization should of course only happen once.
Once W1 is created, you can use it in your model like a normal convolution.