title: mat1 and mat2 shapes cannot be multiplied
body: <p>I am new to AI and python, I'm trying to build an architecture to train a set of images. and later to aim to overfit. but up till now, I couldn't understand how to get the inputs and outputs correctly. I keep seeing the error whenever I try to train the network:</p>
  <p><code>mat1 and mat2 shapes cannot be multiplied (48x13456 and 16x64)</code></p>
  <p>my network:</p>
  <pre><code>net2 = nn.Sequential(
  
  nn.Conv2d(3,8, kernel_size=5, padding=0),
  nn.ReLU(),
  nn.MaxPool2d(kernel_size=2, stride=2),
  
  
  nn.Conv2d(8,16, kernel_size=5, padding=0),
  nn.ReLU(),
  nn.MaxPool2d(kernel_size=2, stride=2),
  
  nn.Flatten(),
  nn.Linear(16,64),
  nn.ReLU(),
  nn.Linear(64,10)
  )
  </code></pre>
  <p>this is a part of a task I'm working on and I really don't get why it's not running. any hints!</p>