fix 0 tensor in readme (#721)

Author: wangna11BD

education/第二天作业.py

@@ -159,7 +159,7 @@ def forward(self, x):
         optimizerD.clear_grad()
 
         errD = errD_real + errD_fake
-        losses[0].append(errD.numpy()[0])
+        losses[0].append(float(errD))
 
         ############################
         # (2) Update G network: maximize log(D(G(z)))
@@ -174,7 +174,7 @@ def forward(self, x):
         optimizerG.step()
         optimizerG.clear_grad()
 
-        losses[1].append(errG.numpy()[0])
+        losses[1].append(float(errG))
 
 
         ############################
@@ -196,7 +196,7 @@ def forward(self, x):
                     plt.xticks([])
                     plt.yticks([])
                     plt.subplots_adjust(wspace=0.1, hspace=0.1)
-                msg = 'Epoch ID={0} Batch ID={1} \n\n D-Loss={2} G-Loss={3}'.format(pass_id, batch_id, errD.numpy()[0], errG.numpy()[0])
+                msg = 'Epoch ID={0} Batch ID={1} \n\n D-Loss={2} G-Loss={3}'.format(pass_id, batch_id, float(errD), float(errG))
                 print(msg)
                 plt.suptitle(msg,fontsize=20)
                 plt.draw()