From a9d5133207d612859574704afc077bcef8bb9046 Mon Sep 17 00:00:00 2001
From: Nikita Shulga <2453524+malfet@users.noreply.github.com>
Date: Wed, 29 Nov 2023 19:27:52 +0000
Subject: [PATCH] [ez][doc] Fix sample code in onnx_dynamo.rst (#114770)

By adding `import torch.nn as nn`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/114770
Approved by: https://github.com/atalman, https://github.com/thiagocrepaldi
---
 docs/source/onnx_dynamo.rst | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docs/source/onnx_dynamo.rst b/docs/source/onnx_dynamo.rst
index 09a09bc3a300ae..be6b9d48a8d399 100644
--- 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):