温馨提示×

pytorch画图如何绘制简单曲线

小樊
81
2024-12-25 15:42:19
栏目: 深度学习

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

0