Update NeuralNet_2_0.py (microsoft#521)

Fixed batch_a not returned in the forward function.
XerxeLee · Jul 31, 2020 · 4cd094f · 4cd094f
1 parent dfe03a1
commit 4cd094f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/A-基础教程/A2-神经网络基本原理简明教程/SourceCode/ch09-NonLinearRegression/HelperClass2/NeuralNet_2_0.py b/A-基础教程/A2-神经网络基本原理简明教程/SourceCode/ch09-NonLinearRegression/HelperClass2/NeuralNet_2_0.py
@@ -54,6 +54,7 @@ def forward(self, batch_x):
             self.A2 = self.Z2
         #end if
         self.output = self.A2
+        return self.output
 
     def backward(self, batch_x, batch_y, batch_a):
         # 批量下降，需要除以样本数量，否则会造成梯度爆炸