pytorch画图如何绘制简单曲线

在PyTorch中，可以使用matplotlib库来绘制简单的曲线。以下是一个示例代码：

import torch
import matplotlib.pyplot as plt

# 创建一个简单的数据集
x = torch.linspace(0, 10, 100)
y = 2 * x + 1

# 绘制曲线
plt.plot(x.numpy(), y.numpy())
plt.xlabel('x')
plt.ylabel('y')
plt.title('Simple Curve')
plt.show()

在这个示例中，我们首先导入了所需的库，然后创建了一个简单的数据集，其中x是从0到10的等间距张量，y是2倍的x加1。接下来，我们使用plt.plot()函数绘制曲线，并将x和y转换为NumPy数组。最后，我们添加了轴标签和标题，并使用plt.show()显示图形。