如何自己创建一个简单的神经网络
MoXing: 模型定义你想要的神经网络模型self.hid1 = torch.nn.Linear(shuru,59)#定义第一个隐藏层(可无)#第一个参数输入(一般为上一层神经元数),这里shuru就是刚才定义的输入数#第一个参数输出(一般为下一层神经元数)self.hid2 = torch.nn.Linear(59, 59)#第二层(可无),同理torch.nn.init.xavier_uniform_(self.hid1.weight)#第一层w。
原创
2024-04-24 18:02:35 ·
405 阅读 ·
1 评论