save_image¶
- torchvision.utils.save_image(tensor: Union[Tensor, List[Tensor]], fp: Union[str, Path, BinaryIO], format: Optional[str] = None, **kwargs) None [原始碼]¶
將給定的 Tensor 儲存到圖片檔案中。
- 參數:
tensor (Tensor 或 list) – 要儲存的圖片。 如果給定一個小批量 tensor,則透過呼叫
make_grid
將 tensor 儲存為圖片網格。fp (字串 或 檔案物件) – 檔案名稱或檔案物件
format (Optional) – 如果省略,則使用的格式由檔案名稱副檔名決定。 如果使用檔案物件而不是檔案名稱,則應始終使用此參數。
**kwargs – 其他參數記錄在
make_grid
中。