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将 tensor 保存为图像网格。fp (string 或 file object) – 文件名或文件对象
format (Optional) – 如果省略，则根据文件名扩展名确定要使用的格式。如果使用文件对象而不是文件名，则应始终使用此参数。
**kwargs – 其他参数记录在
make_grid中。
