2021年3月6日
Pytorch实现两层神经网络
摘要： Pytorch搭建模型 神经网络的基本流程可以分为两大步骤： 网络结构搭建+参数的梯度更新（前向传播+计算参数的梯度+梯度更新） numpy实现 import numpy as np # N是训练的batch size; D_in 是input输入数据的维度; # H是隐藏层的节点数; D_out 阅读全文
posted @ 2021-03-06 21:11 当康 阅读(756) 评论(0) 推荐(0) 编辑