快捷方式

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 (Tensorlist) – 要儲存的影像。如果給定的是小批次 Tensor,則透過呼叫 make_grid 將其儲存為影像網格。

  • fp (stringfile object) – 檔名或檔案物件

  • format (Optional) – 如果省略,使用的格式將根據檔名字尾確定。如果使用的是檔案物件而不是檔名,則應始終使用此引數。

  • **kwargs – 其他引數在 make_grid 中有說明。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源