x2paddle/project_convertor/pytorch/torch2paddle/varbase.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import paddle
-from paddle.fluid.core import VarBase
+from paddle.framework import core
 from x2paddle.utils import paddle_dtypes
 
 
@@ -24,7 +24,7 @@ def is_condition_one(idx):
     a[mask, :]
     a[mask, ...]
     """
-    if not (isinstance(idx[0], paddle.Tensor) and \
+    if not (isinstance(idx[0], paddle.Tensor) and
             idx[0].dtype == paddle_dtypes.t_bool):
         return False
     if len(idx) == 1:
@@ -53,6 +53,8 @@ def is_condition_two(idx):
     return False
 
 
+VarBase = core.eager.Tensor
+
 VarBase.tmp = VarBase.__getitem__