pytorch学习笔记
pytorch一般框架 import torch N, D_in, H, D_out = 64, 1000, 100, 10 #64个数据,输入1000维,中间层100维,输出10维 # 1. 确定训练数据 x = torch.randn(N, D_in) y = torch.randn(N, D_out) # 2. 定义模型 class TwoLayerNet(torch.nn.Module): #在init里定义模型的框架 def __init__(self, D_in, H, D_out): super(TwoLayerNet, self).__init
暂无评论