快捷鍵

save_image

torchvision.utils.save_image(tensor: Union[Tensor, List[Tensor]], fp: Union[str, Path, BinaryIO], format: Optional[str] = None, **kwargs) None[原始碼]

將給定的 Tensor 儲存到圖片檔案中。

參數:
  • tensor (Tensorlist) – 要儲存的圖片。 如果給定一個小批量 tensor,則透過呼叫 make_grid 將 tensor 儲存為圖片網格。

  • fp (字串檔案物件) – 檔案名稱或檔案物件

  • format (Optional) – 如果省略,則使用的格式由檔案名稱副檔名決定。 如果使用檔案物件而不是檔案名稱,則應始終使用此參數。

  • **kwargs – 其他參數記錄在 make_grid 中。

文件

取得 PyTorch 的全面開發者文件

查看文件

教學

取得針對初學者和進階開發者的深入教學

查看教學

資源

尋找開發資源並獲得解答

查看資源