code: def image2np(image:Tensor)->np.ndarray:
      "Convert from torch style `image` to numpy/matplotlib style."
      res = image.cpu().permute(1,2,0).numpy()
      return res[...,0] if res.shape[2]==1 else ref image2np(image:Tensor)->np.ndarray:
      "Convert from torch style `image` to numpy/matplotlib style."
      res = image.cpu().permute(1,2,0).numpy()
      return res[...,0] if res.shape[2]==1 else res
docs: Convert from torch style `image` to numpy/matplotlib style.