在numpy中实现反向传播

时间:2019-12-08 18:06:35

标签: numpy pytorch

我试图通过定义一个函数执行numpy的backprop,该函数在给定输入,权重矩阵和bias的情况下执行某种操作,并使用向后函数返回输出,该函数可用于更新权重。

当前这是我的code,但是我认为推导中存在一些错误,因为W1矩阵的梯度太大。这是与参考torch相同的pytorch实现。

感谢您的帮助。

0 个答案:

没有答案