Linear layer requires 2D tensor. Does it mean that I got to use ‘view’ function to reshape the previous output every time I call it?
But for some experiments BN and RELU are both needed afterwards. They both require 4D, for which I need to call reshape function again and again…
and I can’t find one corresponding reshape function for nn.Sequence()? Is there any plan to add it?