diff --git a/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py b/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py index c1596769bbc6fab1436db401e5e8e68e22ffd835..166653ee1b7c90c6261041e883f6fc1efebe8e96 100644 --- a/x2paddle/project_convertor/pytorch/torch2paddle/varbase.py +++ b/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 @@ -94,9 +94,9 @@ def __getitem__(self, idx): else: return out +VarBase = core.eager.Tensor VarBase.__getitem__ = __getitem__ - VarBase.setitem_tmp = VarBase.__setitem__