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