在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()
函数显示图形。