pub fn save_image<P: AsRef<Path>>(img: &Tensor, p: P) -> Result<()>
Saves an image to disk using the image crate, this expects an input with shape (c, height, width).