溫馨提示×

pytorch畫圖如何設置坐標軸范圍

小樊
96
2024-12-25 15:46:19
欄目: 深度學習

在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()函數顯示圖形。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女