title: Calculating Variance in Pytorch on Spatial axis
body: <p>I am trying to calculate variance in Pytorch but unable to do on multiple axis.</p>
  
  <p>I have similar thing done in Tensorflow but unable to do it on Pytorch as torch.var function takes int as dimension instead of axes.
  <em>Below code is channel last code, I expect axes=[2,3]</em> </p>
  
  <pre><code>Lambda(lambda x: tf.nn.moments(x, axes=[1, 2]))
  </code></pre>
  
  <p>For example, if input_dims = (5, 10, 25, 25) then output_dims should be (5,10, 1, 1). </p>