permuting axes of MLShapedArray ?

I have converted an UIImage to MLShapedArray and by default this is NCHW format. I need to permute it into NCWH to prepare it for an ML model. What is the recommended way to achieve this ?

permuting axes of MLShapedArray ?
 
 
Q