在PyTorch中，可以使用torchvision.utils.make_grid()函數進行多圖繪制。這個函數可以將多個圖像拼接成一個網格圖。以下是一個簡單的示例：
首先，確保已經安裝了torchvision庫。如果沒有安裝，可以使用以下命令安裝：
pip install torchvision
然后，可以使用以下代碼進行多圖繪制：
import torch
from torchvision import utils
import matplotlib.pyplot as plt
# 創建一些示例圖像
image1 = torch.randn(3, 256, 256)
image2 = torch.randn(3, 256, 256)
image3 = torch.randn(3, 256, 256)
# 將圖像轉換為張量
tensor1 = image1.unsqueeze(0)
tensor2 = image2.unsqueeze(0)
tensor3 = image3.unsqueeze(0)
# 使用make_grid()函數將圖像拼接成一個網格圖
grid = utils.make_grid([tensor1, tensor2, tensor3], nrow=1, normalize=True)
# 使用matplotlib庫繪制網格圖
plt.imshow(grid[0].numpy().transpose((1, 2, 0)))
plt.axis('off')
plt.show()
在這個示例中，我們首先創建了三個隨機圖像，然后使用unsqueeze()函數將它們轉換為形狀為(1, 3, 256, 256)的張量。接下來，我們使用utils.make_grid()函數將這些圖像拼接成一個網格圖，其中nrow=1表示將圖像水平排列。最后，我們使用matplotlib庫繪制網格圖。
