[Dy2stat] Fix to_tensor Bug Reported from QA

python/paddle/fluid/dygraph/dygraph_to_static/basic_api_transformer.py

@@ -33,10 +33,11 @@ def __init__(self, wrapper_root):
         self.root = wrapper_root.node
         self.class_node_dict = {}
 
-        self.name_to_tensor_shape = {}
-
     def transform(self):
+        to_tensor_transformer = ToTensorTransformer(self.root)
+        to_tensor_transformer.transform()
         self.visit(self.root)
+
         return self.wrapper_root
 
     def visit_Assign(self, node):
@@ -62,11 +63,6 @@ def visit_Expr(self, node):
 
     def _visit_Call(self, node):
         assert isinstance(node, gast.Call)
-        # Replace API `to_variable` with `fluid.layers.assign`
-        if is_to_variable(node):
-            node = to_assign_node(node)
-            return node
-
         func_name = astor.to_source(gast.gast_to_ast(node.func))
 
         if self._is_dygraph_forward(func_name):
@@ -102,6 +98,29 @@ def _update_class_node_dict(self, node):
         return False
 
 
+class ToTensorTransformer(gast.NodeTransformer):
+    """
+    Class to transform paddle.to_tensor and paddle.to_variable to paddle.assign
+    """
+
+    def __init__(self, node):
+        assert isinstance(
+            node, gast.AST
+        ), "Input non-gast.AST node for the initialization of ToTensorTransformer."
+        self.root = node
+
+    def transform(self):
+        self.visit(self.root)
+        return self.root
+
+    def visit_Call(self, node):
+        assert isinstance(node, gast.Call)
+        if is_to_variable(node):
+            node = to_assign_node(node)
+        self.generic_visit(node)
+        return node
+
+
 def is_to_variable(node):
     assert isinstance(node, gast.Call)
     api_name = utils.ast_to_source_code(node.func).strip()

python/paddle/fluid/tests/unittests/dygraph_to_static/test_basic_api_transformation.py

@@ -64,13 +64,11 @@ def dyfunc_int_to_tensor(x):
 
 
 def dyfunc_float_to_tensor(x):
-    res = paddle.to_tensor(2.0)
-    return res
+    return paddle.to_tensor(2.0)
 
 
 def dyfunc_bool_to_tensor(x):
-    res = paddle.to_tensor(True)
-    return res
+    return paddle.to_tensor(True)
 
 
 class TestDygraphBasicApi_ToVariable(unittest.TestCase):