一、函数解释 如果输入x,输出是y,则求y关于x的导数(梯度): def grad(outputs, inputs, grad_outputs=None, retain_graph=None, create_graph=False, only_inputs=True, allow_unused=False): rComputes and returns the sum of gradients of outputs ...
通过一个小案例,反映出DataLoader使用完整流程!! import torch import torch.utils.data as Data BATCH_SIZE = 3 #批训练数据个数 x = torch.linspace(1,10,10) #x data (torch tensor) y = torch.linspace(10,1,10) #y data (torch tensor) #随后我们需要把X和Y组成一个完整的数据集,并转 ...