胸毛飘荡的博客

import timeimport torchimport torch.nn as nnfrom torch.autograd import Variableimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as pltEPOCH = 1 ...

import torchfrom torch.autograd import Variableimport torch.nn.functional as Fimport matplotlib.pyplot as plt# 从-1到1均匀生成10个数据tensor_x = torch.linspace(-1, 1, 100)'''这里是把一维矩阵转二维矩阵，dim=1表示从tenso...