save_image¶
- torchvision.utils.save_image(tensor: Union[Tensor, List[Tensor]], fp: Union[str, Path, BinaryIO], format: Optional[str] = None, **kwargs) None[源代码]¶
将给定的张量保存到图像文件。
- 参数：
tensor (Tensor 或 列表) – 要保存的图像。如果给定一个小批次张量，则通过调用
make_grid将张量保存为图像网格。fp (字符串 或 文件对象) – 文件名或文件对象
format (可选) – 如果省略，则根据文件名扩展名确定要使用的格式。如果使用文件对象而不是文件名，则应始终使用此参数。
**kwargs – 其他参数在
make_grid中有说明。
