torch에서 (c,w,h)순서 바꿔줘야 할 때 permute(1,2,0)하면 (w,h,c)로 돌아옴.