consistent nn. and nn.functional: part 5 docs (#12161)

2021-06-14 13:34:32 -07:00
parent 88e84186e5
commit 040283170c
5 changed files with 9 additions and 9 deletions
--- a/docs/source/add_new_model.rst
+++ b/docs/source/add_new_model.rst
@@ -518,7 +518,7 @@ PyTorch, called ``SimpleModel`` as follows:

 .. code:: python

-   import torch.nn as nn
+   from torch import nn

   class SimpleModel(nn.Module):
       def __init__(self):