Some confusions about nn.Linear that bothered me ;-(
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?
hbkunn
Atcold
apaszke
thank you for replying me! I’m so sorry for not noticing BatchNorm1d in the document. Now my confusion is perfectly solved~
hbkunn
that’s reasonable enough~ Thanks for your replying!
hbkunn