torch画散点图

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，还可以计算梯度。