保存图片

torchvision.utils.save_image(tensor: Union[Tensor, List[Tensor]], fp: Union[str, Path, BinaryIO], format: Optional[str] = None, **kwargs) → None

将给定的Tensor保存为图像文件。

Parameters:

• 张量 (张量或列表) – 要保存的图像。如果给定一个小批量张量，通过调用make_grid将张量保存为图像网格。

• fp (字符串 或 文件对象) – 文件名或文件对象

• format (可选) – 如果省略，则根据文件扩展名确定使用的格式。 如果使用文件对象而不是文件名，则应始终使用此参数。

• **kwargs – 其他参数在make_grid中有详细说明。