fix example

Author: xadupre

_doc/examples/plot_export_with_auto.py

@@ -46,7 +46,7 @@ def forward(self, x, y, z):
 # %%
 # Then we could make it a different one.
 
-dz = torch.export.Dim("dz") * 2
+dz = torch.export.Dim("dz")
 try:
     ep = torch.export.export(
         model,
@@ -84,13 +84,12 @@ def forward(self, x, y, z):
 # The same result can be obtained with ``torch.export.Dim.AUTO``.
 
 AUTO = torch.export.Dim.AUTO
-print(
-    torch.export.export(
-        model,
-        (x, y, z),
-        dynamic_shapes=({0: AUTO, 1: AUTO}, {0: AUTO, 1: AUTO}, {0: AUTO, 1: AUTO}),
-    )
+ep = torch.export.export(
+    model,
+    (x, y, z),
+    dynamic_shapes=({0: AUTO, 1: AUTO}, {0: AUTO, 1: AUTO}, {0: AUTO, 1: AUTO}),
 )
+print(ep)
 
 # %%