PyTorch Forums
How to reshape a tensor back to it previous shape after using view()
vision
my3bikaht
(Sergey)
February 7, 2022, 2:44pm
3
112*122 = 12544
16053760 / 80 / 16 = 12542
You’re probably lost 1 padding in conv layer before.
show post in topic