在PyTorch中，可以使用Matplotlib庫來繪制圖形并添加標注信息。以下是一個簡單的示例，展示了如何在繪制直線時添加標注信息：
import torch
import matplotlib.pyplot as plt
# 創建數據
x = torch.tensor([0, 1, 2, 3, 4])
y = torch.tensor([0, 1, 4, 9, 16])
# 繪制直線
plt.plot(x, y, label='y = x^2')
# 添加標注信息
plt.annotate('頂點', xy=(2, 4), xytext=(3, 5), arrowprops=dict(facecolor='black', shrink=0.05))
plt.xlabel('x軸')
plt.ylabel('y軸')
plt.legend()
# 顯示圖形
plt.show()
在這個示例中，我們首先導入了所需的庫，然后創建了兩個張量x和y來表示數據點。接下來，我們使用plt.plot()函數繪制了一條直線，并通過label參數添加了圖例信息。然后，我們使用plt.annotate()函數在直線上添加了一個標注信息，指定了標注的位置和文本內容。最后，我們設置了坐標軸標簽和圖例，并使用plt.show()函數顯示了圖形。
你可以根據自己的需求修改這個示例，以適應不同的繪圖需求。
