温馨提示×

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

小樊
83
2024-12-25 15:46:19
栏目: 深度学习
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:matlab画图如何设置坐标轴

0