I don’t think you can. If you wanted to do that, you could define these four 1x2 parameters separately and use torch.cat to combine them during the forward. (Or use size 2 and use torch.stack.)
Best regards
Thomas