title: How can I get the MSE of a tensor across a specific dimension?
body: <p>I have 2 tensors with <code>.size</code> of <code>torch.Size([2272, 161])</code>. I want to get mean-squared-error between them. However, I want it along each of the 161 channels, so that my error tensor has a <code>.size</code> of <code>torch.Size([161])</code>. How can I accomplish this?</p>
  
  <p>It seems that <code>torch.nn.MSELoss</code> doesn't let me specify a dimension.</p>