Patched python level LoDTensor

PaddlePaddle · Jan 12, 2022 · 064eb19 · 064eb19
1 parent 786df9a
commit 064eb19
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py
@@ -32,6 +32,10 @@
     except Exception as e:
         raise e
 
+# Patch LoDTensor
+import core
+core.LoDTensor = core.Tensor
+
 # import all class inside framework into fluid module
 from . import framework
 from .framework import *
@@ -70,9 +74,6 @@
 from .param_attr import ParamAttr, WeightNormParamAttr
 from .data_feeder import DataFeeder
 
-import core
-core.LoDTensor = core.Tensor
-
 from .core import LoDTensorArray, Scope, _Scope
 from .core import CPUPlace, XPUPlace, CUDAPlace, CUDAPinnedPlace, NPUPlace, IPUPlace, MLUPlace
 from .incubate import fleet