title: Change Image format from NHWC to NCHW for Pytorch
body: <p>In pytorch we need images in <code>NCHW</code> format but my images are <code>NHWC</code>.</p>
  
  <p>What is the procedure to feed this image to CNN? (I have found <a href="https://discuss.pytorch.org/t/swap-axes-in-pytorch/970" rel="noreferrer">this solution</a> which suggests to use "permute" but where and how should i use it?)</p>