Update extractor.py

Author: fangfangssj

graph_net/torch/extractor.py

@@ -4,6 +4,7 @@
 from typing import Union, Callable
 from . import utils
 
+torch._dynamo.config.capture_scalar_outputs = True
 
 def extract(name, dynamic=True, mut_graph_codes=None, placeholder_auto_rename=False):
     """