What if I want to assign to Conv2d layer’ weights instead  nn.Linear layer? Is it still possible to update the a and b?
As you said, in linear layer, we can apply torch.matmul in forward(),
How about using the conv2d operator with initialize weight by torch.matmul(a,b), can i also do something similar?