码迷,mamicode.com
首页 > 其他好文 > 详细

torch画散点图

时间:2018-12-25 11:25:08      阅读:124      评论:0      收藏:0      [点我收藏+]

标签:code   lin   包含   port   lib   技术分享   ima   nump   img   

import torch
from torch.autograd import Variable
import torch.nn.functional as F
import matplotlib.pyplot as plt
x=torch.unsqueeze(torch.linspace(-1,1,100),dim=1)   #torch.linspace本身是一维向量,unsqueeze是增加维度,把一维化为二维
y=x.pow(2)+0.2*torch.rand(x.size())   
#0.2*torch.rand(x.size())相当于给散点图加噪声
x,y=Variable(x),Variable(y)
plt.scatter(x.data.numpy(),y.data.numpy())
plt.show()

输出结果为:

技术分享图片

代码中,Variable是变量的意思。包含很多属性,常用的是.data,还可以计算梯度。

torch画散点图

标签:code   lin   包含   port   lib   技术分享   ima   nump   img   

原文地址:https://www.cnblogs.com/wmy-ncut/p/10172355.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!