摘要:
pytorch基础知识 import torch x = torch.tensor([2,3,4], dtype=torch.float) # 创建一个Tensor,值为[2.,3.,4.],类型为 float # 创建一个需要求 梯度的 tensor。 x2 = torch.tensor([2,3 阅读全文
摘要:
#coding=utf-8import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd import Variableclass Net(nn.Module): #定义Net的初始化函数,这个函 阅读全文