save_image¶
- torchvision.utils.save_image(tensor: Union[Tensor, List[Tensor]], fp: Union[str, Path,BinaryIO], format: Optional[str] = None, **kwargs) None[source]¶
将给定 Tensor 保存到图像文件中。
- 参数：
tensor (Tensor 或 list) – 要保存的图像。如果给定的是小批量 Tensor，则通过调用
make_grid将其保存为图像网格。fp (string 或 file object) – 文件名或文件对象
format (Optional) – 如果省略，使用的格式将根据文件名后缀确定。如果使用的是文件对象而不是文件名，则应始终使用此参数。
**kwargs – 其他参数在
make_grid中有说明。
