fix bug #79

ShusenTang · Dec 30, 2019 · b3401dd · b3401dd
1 parent cd848d2
commit b3401dd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/chapter05_CNN/5.8_nin.md b/docs/chapter05_CNN/5.8_nin.md
@@ -43,6 +43,7 @@ NiN是在AlexNet问世不久后提出的。它们的卷积层设定有类似之
 
 ``` python
 # 已保存在d2lzh_pytorch
+import torch.nn.functional as F
 class GlobalAvgPool2d(nn.Module):
     # 全局平均池化层可通过将池化窗口形状设置成输入的高和宽实现
     def __init__(self):