My image size is (136,136,3)
After applying x.view(x.size(0),-1) torch.size(136,408)
How to convert my image to (1,3,136,136)
If I use torch.rand(1,3,136,136) I got an other error  size mismatch , m1: [1×2] , m2: [580×80]
Error at line y=F.relu(self.layer17(y))