[ez][doc] Fix sample code in onnx_dynamo.rst (pytorch#114770)

By adding `import torch.nn as nn` Pull Request resolved: pytorch#114770 Approved by: https://github.com/atalman, https://github.com/thiagocrepaldi
kurtamohler · Nov 29, 2023 · a9d5133 · a9d5133
1 parent ffa974b
commit a9d5133
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/onnx_dynamo.rst b/docs/source/onnx_dynamo.rst
@@ -53,6 +53,7 @@ See below a demonstration of exporter API in action with a simple Multilayer Per
 .. code-block:: python
 
   import torch
+  import torch.nn as nn
 
   class MLPModel(nn.Module):
     def __init__(self):