温馨提示×

pytorch画图如何设置坐标轴范围

小樊
82
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