title: PyTorch - unable to use batchnorm1d with Linear
body: <p>Using PyTorch's <code>BatchNorm1D</code> on a 1-D tensor gives the error:</p>
  
  <blockquote>
    <p>RuntimeError: running_mean should contain 1 elements not 2304</p>
  </blockquote>
  
  <p>Any suggestions on what might be wrong?</p>
  
  <p><strong>My Code:</strong></p>
  
  <pre><code>self.net_common = nn.Sequential (
      nn.Linear(64*64, 48*48),
      nn.BatchNorm1d(48*48),
      nn.Tanh(),
      nn.Dropout(p=0.25),
      nn.Linear(48*48, 32*32),
      nn.BatchNorm1d(32*32),
      nn.Tanh(),
  )
  </code></pre>