返回顶部
摘要: pytorch基础知识 import torch x = torch.tensor([2,3,4], dtype=torch.float) # 创建一个Tensor,值为[2.,3.,4.],类型为 float # 创建一个需要求 梯度的 tensor。 x2 = torch.tensor([2,3 阅读全文
posted @ 2021-11-03 21:06 tianyudizhua 阅读(44) 评论(0) 推荐(0) 编辑
摘要: #coding=utf-8import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd import Variableclass Net(nn.Module): #定义Net的初始化函数,这个函 阅读全文
posted @ 2021-11-03 19:28 tianyudizhua 阅读(333) 评论(0) 推荐(0) 编辑