code: def nhwc(
      x: Union[np.ndarray, JaxArray]
  ) -> Union[np.ndarray, JaxArray]:
      """Converts an array in (N,C,H,W) format to (N,H,W,C) format."""
      dims = list(range(x.ndim))
      dims.append(dims.pop(-3))
      return x.transpose(dims)
  
file_path: objax/objax/util/image.py
imports: import io
  from typing import Union, BinaryIO, IO
  import jax.numpy
  import numpy
  from PIL import Image
  from objax.typing import JaxArray
label_desc: Use image node_id 4 for calling a global function with example usage: nhwc(x) and returns: x