Hi,

I am ignoring the bias term here, I just want a weight term. So, I think according to the docs the second param of F.linear must be a matrix W with (out_dim, in_dim) right?
Oh, so it is of no use if I don’t define self.w = nn.Parameter(torch.randn(row, col))?
Actually, my whole idea behind doing this was that I do not want to manually set a number for the output of this model since that would be anyway defined by the number of pixels that would be in my feature map just before I flatten the feature map. So, that is why I wanted to make my model code agnostic of that. Sorry if I am not clear… Please let me know if I am not clear and if there is some way to handle this… - I mean if I could just forget all this hassle and just type in numbers for whatever dimensions that I need but wanted a little better way to do that