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

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