Can nn.Linear() work with variable-sized inputs?
I have an input of shape (2048, x) and would like to convert it to be of shape (2048) using a fully-connected layer – is this possible?
When I passed in x as an argument to the model, the only issue I had was with saving/loading the weights since x would change the number of parameters in the layer. Any ideas about how this issue can be resolved?