在PyTorch中,可以使用matplotlib庫來繪制圖形并設置坐標軸范圍
import torch
import matplotlib.pyplot as plt
# 創建一個簡單的數據集
x = torch.linspace(0, 10, 100)
y = torch.sin(x)
# 繪制圖形
plt.plot(x, y)
# 設置坐標軸范圍
plt.xlim(0, 10) # 設置x軸范圍
plt.ylim(-1, 1) # 設置y軸范圍
# 顯示圖形
plt.show()
在這個示例中,我們首先導入了所需的庫,然后創建了一個簡單的數據集。接下來,我們使用plt.plot()
函數繪制圖形。為了設置坐標軸范圍,我們分別調用plt.xlim()
和plt.ylim()
函數,傳入我們想要的范圍。最后,我們使用plt.show()
函數顯示圖形。