title: RuntimeError: size mismatch, m1: [28 x 28], m2: [784 x 128]
body: <p>After training my model, I tried to plot graph of the softmax output, but it resulted in the runtime error mentioned in the title.</p>
  
  <p>Here is the following code snippet:</p>
  
  <pre><code>%matplotlib inline
  %config InlineBackend.figure_format = 'retina'
  
  import helper
  
  # Test out your network!
  
  dataiter = iter(testloader)
  images, labels = dataiter.next()
  img = images[1]
  
  # TODO: Calculate the class probabilities (softmax) for img
  ps = torch.exp(model(img))
  
  # Plot the image and probabilities
  helper.view_classify(img, ps, version='Fashion')
  </code></pre>