qq_43251445的博客

import torch import torch.optim import matplotlib.pyplot as plt import torch.nn.functional as f from torch.autograd import Variable # unsequeeze(param) 增加param的维度，在torch中处理2维的张量 x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1) # 目标函数表达式 y = x.pow(2)

新版直接传入参数Data.TensorDataset（x, y） 没有GPU的num_workers=2会报错，默认使用0。num_workers：用于数据加载的子进程数目，默认0 import torch import torch.utils.data as Data # 批训练数据的大小 BATCH_SIZE = 5 x = torch.linspace(1, 10, 10) y = torch.linspace(10, 1, 10) # 新版中TensorDataset直接传入参数即可 # .