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中有說明。