Has this been included in official modules?
It could be useful because you decouple your module from the forward method. I mean, if I wanna convert FC to conv layers on a pre-trained model, and the forward method includes the x.view call, I can’t, right?
With the flatten operator inside the model, I can just unplug it and put a conv before. Does this make sense or am I looking at it from the wrong side?