save_image

torchvision.utils.save_image(tensor: Union[Tensor, list[torch.Tensor]], fp: Union[str, Path, BinaryIO], format: Optional[str] = None, **kwargs) → None

將給定的 Tensor 儲存到影像檔案。

引數:

• tensor (Tensor 或 list) – 要儲存的影像。如果給定一個 mini-batch tensor，則透過呼叫 make_grid 將 tensor 儲存為影像網格。

• fp (string 或 檔案物件) – 檔名或檔案物件

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

• **kwargs – 其他引數在 make_grid 中進行了文件記錄。